Merge pull request #11 from miguelcarcamov/development

CHG: reverting changes
miguelcarcamov · Oct 30, 2022 · 7b8c9b9 · 7b8c9b9
2 parents e467472 + 126bc19
commit 7b8c9b9
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/snow/imaging/gpuvmem.py b/src/snow/imaging/gpuvmem.py
@@ -271,7 +271,7 @@ def __restore_pyra(
         )
         image = dirty_mapper.transform()[0].data[0].compute()
 
-        fits_io.write(image, output_name=residual_casa_image + ".fits", header=header_model)
+        fits_io.write(image, output_name=residual_casa_image + ".fits")
 
         psf = dataset.psf[0]
 
@@ -287,7 +287,7 @@ def __restore_pyra(
 
         im_restored = im_model_convolved + image
 
-        fits_io.write(im_restored, output_name=restored_image + ".fits", header=header_model)
+        fits_io.write(im_restored, output_name=restored_image + ".fits")
 
         return residual_casa_image + ".fits", restored_image + ".fits"