Small corrections and testing the convertor

resolve/ubik_tools/fits.py

@@ -118,10 +118,12 @@ def field2fits(field, file_name, sky_dom_dict=None, observations=[], header_over
     if direction is not None:
         h["EQUINOX"] = direction.equinox
 
+    field = field.val if nifty_field else field
+
     for kk, vv in header_override.items():
         h[kk] = vv
-    for t_ind, t_val in (t_coord):
-        val = field.val[:, t_ind]  # Select time
+    for t_ind, t_val in enumerate(t_coord):
+        val = field[:, t_ind]  # Select time
         val = np.transpose(val, (0, 1, 3, 2))  # Switch spatial axes
         val = val.astype(np.float32)
         val = np.require(val, dtype=np.float32, requirements="F")