WIP use smooth piff models (#27)

des-science · Mar 19, 2024 · cc0df22 · cc0df22
1 parent a354c19
commit cc0df22
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/montara/des_tile.py b/montara/des_tile.py
@@ -470,7 +470,7 @@ def setup(self, config, base, file_num, logger):
 
             # make sure the draw method is correct for PSFEx
             assert base['stamp']['draw_method'] == 'no_pixel'
-        elif base["psf"]["type"] == "DES_Piff":
+        elif base["psf"]["type"] in ["DES_Piff", "DES_SmoothPiff"]:
             output_piff_path = os.path.join(
                 base["base_dir"],
                 os.path.relpath(orig_piff_path, imsim_data),
@@ -481,7 +481,7 @@ def setup(self, config, base, file_num, logger):
             shutil.copyfile(orig_piff_path, output_piff_path)
 
             # make sure the draw method is correct for Piff
-            if base["psf"].get("depixelize", False):
+            if base["psf"].get("depixelize", False) or base["psf"]["type"] == "DES_SmoothPiff":
                 assert base['stamp']['draw_method'] == 'auto'
             else:
                 assert base['stamp']['draw_method'] == 'no_pixel'