From 6f56b77d55f0c52655178881ec382c68ef9f76f1 Mon Sep 17 00:00:00 2001
From: Philipp Arras <parras@mpa-garching.mpg.de>
Date: Tue, 24 May 2022 16:59:32 +0200
Subject: [PATCH] Improve field2fits

---
 resolve/ubik_tools/fits.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/resolve/ubik_tools/fits.py b/resolve/ubik_tools/fits.py
index b034ebe7..2c2bbb55 100644
--- a/resolve/ubik_tools/fits.py
+++ b/resolve/ubik_tools/fits.py
@@ -107,4 +107,7 @@ def field2fits(field, file_name, observations=[], header_override={}):
         hdu = pyfits.PrimaryHDU(val, header=h)
         hdulist = pyfits.HDUList([hdu])
         base, ext = splitext(file_name)
-        hdulist.writeto(base + f"time{t_val}" + ext, overwrite=True)
+        if len(tdom.coordinates) == 1:
+            hdulist.writeto(base + ext, overwrite=True)
+        else:
+            hdulist.writeto(base + f"time{t_val}" + ext, overwrite=True)
-- 
GitLab