sso updates for new config arrangement

LSSTDESC · Jul 15, 2024 · bca6a51 · bca6a51
1 parent e43dc18
commit bca6a51
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 6 deletions.
diff --git a/skycatalogs/sso_catalog_creator.py b/skycatalogs/sso_catalog_creator.py
@@ -6,6 +6,7 @@
 import pyarrow as pa
 import pyarrow.parquet as pq
 from .objects.base_object import LSST_BANDS
+from .utils.config_utils import assemble_provenance
 
 """
 Code for creating sky catalogs for sso objects
@@ -184,11 +185,12 @@ def create_sso_catalog(self):
             self._write_hp(h, hps_by_file, arrow_schema)
 
         # Add config information for sso
-        prov = assemble_provenance(self._catalog_creator._pkg_root,
-                                   inputs={'sso_truth': self._sso_truth,
-                                           'sso_sed' : self._sso_sed},
-                                   run_options=self._catalog_creator._run_options)
-        self._catalog_creator.config_writer.write_configs('sso', prov)
+        prov = assemble_provenance(
+            self._catalog_creator._pkg_root,
+            inputs={'sso_truth': self._sso_truth,
+                    'sso_sed' : self._sso_sed},
+            run_options=self._catalog_creator._run_options)
+        self._catalog_creator._config_writer.write_configs('sso', prov)
 
     def _create_sso_flux_pixel(self, pixel, arrow_schema):
         '''

diff --git a/skycatalogs/utils/source_config.py b/skycatalogs/utils/source_config.py
@@ -178,7 +178,7 @@ def create_snana_config(provenance, area_partition=None, data_file_type=None):
 def create_sso_config(provenance, area_partition=None, data_file_type=None):
     opt_dict = {'area_partition': area_partition,
                 'data_file_type': data_file_type}
-    return _generic_create('sso_template_yaml', provenance, opt_dict)
+    return _generic_create('sso_template.yaml', provenance, opt_dict)
 
 
 def _generic_create(template_filename, provenance, opt):