pz-rail-astro-tools 1.0.7__py3-none-any.whl → 1.1.0__py3-none-any.whl

{pz_rail_astro_tools-1.0.7.dist-info → pz_rail_astro_tools-1.1.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.2
 Name: pz-rail-astro-tools
-Version: 1.0.7
+Version: 1.1.0
 Author-email: "LSST Dark Energy Science Collaboration (DESC)" <lsst-desc-rail-admin@slac.stanford.edu>
 License: MIT License
         

{pz_rail_astro_tools-1.0.7.dist-info → pz_rail_astro_tools-1.1.0.dist-info}/RECORD

@@ -1,11 +1,11 @@
 rail/astro_tools/__init__.py,sha256=4J-E0Ug0RP3Y38g8kJ6auAPDPEI4Osf19tuSOAACfxk,432
-rail/astro_tools/_version.py,sha256=B_2jTRPLk-a7dhEz8Rto7VRGwxfpXrqO4mYrudjn7bw,411
+rail/astro_tools/_version.py,sha256=CqDGE4B1ZqZ-56mxeOFcXRTmlxrdOh4ayrjbcPjziE4,411
 rail/creation/degraders/grid_selection.py,sha256=gJKM0AYdG-HgvjTS3ovupbyoB5bxKhZn9FAAb94pQXY,12429
 rail/creation/degraders/observing_condition_degrader.py,sha256=y6Nal_kSdAz6N8amY4oV6mQ3upa2nIgVbV9P6F0PKwA,19977
-rail/creation/degraders/photometric_errors.py,sha256=xeLI5dH4P-kseClMsy7vTes6EBXh2Ca8_-9-nnFt3b0,4396
+rail/creation/degraders/photometric_errors.py,sha256=jx7JL5-lzKm6IbNTYwHqkEeGZduXC59Tad2UNwEJWlc,4703
 rail/creation/degraders/spectroscopic_degraders.py,sha256=27kylHksXf4MhgdfefSL7C2S3WCK-0YQ1Oc8dOHIBhc,5350
 rail/creation/degraders/spectroscopic_selections.py,sha256=TXlghBWLAniokbUpgw4IQVf3y3R6BwralS0ddiVgMp4,22205
-rail/creation/degraders/unrec_bl_model.py,sha256=YEei4uAdVrWYwO2mhW0KHU2-oUwWaz1F6Ft3UfkJjQ4,7065
+rail/creation/degraders/unrec_bl_model.py,sha256=E89DVV67FcHozCjtUz4ty3Ak3q8H9lnNnh5wNWi_R8M,7476
 rail/creation/engines/gcr_engine.py,sha256=LSp599uQGu8GcFNUmib792ScQwSB_Og8VxPLTt9aSNU,2981
 rail/examples_data/creation_data/data/HSC_grid_settings.pkl,sha256=zpJoSR9TSnDyL64DPXivIMam3fn6GOZoyoJoeBj2LAk,141
 rail/examples_data/creation_data/data/hsc_ratios_and_specz.hdf5,sha256=liH54wuuuHxToqjoAwQ_fFmEefk3haNweVThMwdDKN4,13823712
@@ -14,13 +14,13 @@ rail/examples_data/creation_data/data/survey_conditions/DC2-mask-neg-nside-128.f
 rail/examples_data/creation_data/data/survey_conditions/minion_1016_dc2_Median_airmass_i_and_nightlt1825_HEAL.fits,sha256=twvyce70G3h1MQnGGFQJG9wZc8UbuNqXc6mjtiuaL1s,1581120
 rail/examples_data/creation_data/data/survey_conditions/minion_1016_dc2_Median_fiveSigmaDepth_i_and_nightlt1825_HEAL.fits,sha256=RbMMFSVJQxGhZnL4BnISE9VTNljfTZwV6Bax6NxverY,1581120
 rail/examples_data/testdata/rubin_dm_dc2_example2.pq,sha256=_Iihbbcz1A332QAkaMiFacT4QAbBQBMTOwpfrS7AclE,25884
-rail/pipelines/degradation/apply_phot_errors.py,sha256=vj2avYWxfavORIXzJ7925V9YkuRmRMBSP0C8-DmgkSw,1958
+rail/pipelines/degradation/apply_phot_errors.py,sha256=E6H1Qza1IkO9VWqqxeny6MYkw3gURWscwKxzgUHEOko,2087
 rail/pipelines/degradation/blending.py,sha256=oxZHy1Tai70wT4g4xnBr24z2LxmR1p_J6lMoTN0nFc8,564
 rail/pipelines/degradation/spectroscopic_selection_pipeline.py,sha256=D0sDi6nrasv5CvQkLXgmMp0QbDrhGFRXzt6sWjlbD68,2028
-rail/pipelines/degradation/truth_to_observed.py,sha256=7Mbgxgzn2gn0MCRPyDLk4zQO00liVQNwsfDbmXKrbYM,2635
+rail/pipelines/degradation/truth_to_observed.py,sha256=nfzMKdZd69_P1HnHy56UcjMUpFdRHf6g55oNhlW5Gzg,2995
 rail/tools/photometry_tools.py,sha256=wfKAeMz5T_CYDPC661MHtU1aJXVqomvahZl87KuE7uY,19541
-pz_rail_astro_tools-1.0.7.dist-info/LICENSE,sha256=tLMEN21HbzvT-7umOVVvPjaJZbQxCa-2bAeYLwG5Q04,1102
-pz_rail_astro_tools-1.0.7.dist-info/METADATA,sha256=-QxK7OAGDIX2aGlvh1lsYFwEy-wMmN96H1MubkZGoqA,4033
-pz_rail_astro_tools-1.0.7.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
-pz_rail_astro_tools-1.0.7.dist-info/top_level.txt,sha256=6R6sqn-85I8YZTzmB9gv7sqckB2QaOTHbu-QLGTWBRE,5
-pz_rail_astro_tools-1.0.7.dist-info/RECORD,,
+pz_rail_astro_tools-1.1.0.dist-info/LICENSE,sha256=tLMEN21HbzvT-7umOVVvPjaJZbQxCa-2bAeYLwG5Q04,1102
+pz_rail_astro_tools-1.1.0.dist-info/METADATA,sha256=dt1QkfApkKIjqXp25OoXehGfM7W1_bVhpok1bugejaY,4033
+pz_rail_astro_tools-1.1.0.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+pz_rail_astro_tools-1.1.0.dist-info/top_level.txt,sha256=6R6sqn-85I8YZTzmB9gv7sqckB2QaOTHbu-QLGTWBRE,5
+pz_rail_astro_tools-1.1.0.dist-info/RECORD,,

{pz_rail_astro_tools-1.0.7.dist-info → pz_rail_astro_tools-1.1.0.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (75.6.0)
+Generator: setuptools (75.8.0)
 Root-Is-Purelib: true
 Tag: py3-none-any
 

rail/astro_tools/_version.py

@@ -12,5 +12,5 @@ __version__: str
 __version_tuple__: VERSION_TUPLE
 version_tuple: VERSION_TUPLE
 
-__version__ = version = '1.0.7'
-__version_tuple__ = version_tuple = (1, 0, 7)
+__version__ = version = '1.1.0'
+__version_tuple__ = version_tuple = (1, 1, 0)

rail/creation/degraders/photometric_errors.py

@@ -47,21 +47,27 @@ class PhotoErrorModel(Noisifier):
                 default = val.default
                 
             # Add this param to config_options
-            self.config[key] = Param(
+            # Use setattr() becuase ceci.StageConfig has
+            # implemented __setitem__ to just set the value
+            # rather than add the parameters
+            param = Param(
                 None,  # Let PhotErr handle type checking
                 default,  
                 msg="See the main docstring for details about this parameter.",
                 required=False,
             )
+            setattr(self.config, key, param)
 
     def reload_pars(self, args):
         """ This is needed b/c the parameters are dynamically defined, 
         so we have to reload them _after_ then have been defined """
-        copy_args = args.copy()
         if isinstance(args, dict):
+            # coming from python, add 'config' to the configuration
+            copy_args = args.copy()
             copy_args['config'] = args
         else:  # pragma: no cover
-            copy_args['config'] = vars(args)            
+            # coming from cli, just convert to a dict
+            copy_args = vars(args).copy()
         self.load_configs(copy_args)
         self._io_checked = False
         self.check_io()

rail/creation/degraders/unrec_bl_model.py

@@ -8,6 +8,8 @@ import numpy as np, pandas as pd
 import FoFCatalogMatching
 
 
+lsst_zp_dict = {'u':12.65, 'g':14.69, 'r':14.56, 'i': 14.38, 'z':13.99, 'y': 13.02}
+
 class UnrecBlModel(Degrader):
     """Model for Creating Unrecognized Blends.
 
@@ -22,6 +24,7 @@ class UnrecBlModel(Degrader):
                           dec_label=Param(str, 'dec', msg='dec column name'),
                           linking_lengths=Param(float, 1.0, msg='linking_lengths for FoF matching'),
                           bands=SHARED_PARAMS,
+                          zp_dict=Param(dict, lsst_zp_dict, msg='magnitude zeropoints dictionary'),
                           ref_band=SHARED_PARAMS,
                           redshift_col=SHARED_PARAMS,
                           match_size=Param(bool, False, msg='consider object size for finding blends'),
@@ -99,17 +102,18 @@ class UnrecBlModel(Degrader):
         unique_id = np.unique(group_id)
 
         ra_label, dec_label = self.config.ra_label, self.config.dec_label
-        cols = [ra_label, dec_label] + [b for b in self.config.bands] + self.blend_info_cols
+        cols = [ra_label, dec_label] + [b for b in self.config.bands] + [self.config.redshift_col] + self.blend_info_cols
 
         N_rows = len(unique_id)
         N_cols = len(cols)
 
         # compute the fluxes once for all the galaxies
-        fluxes = {b:10**(-data[b]/2.5) for b in self.config.bands}
+        fluxes = {b:10**(-(data[b] - self.config.zp_dict[b])/2.5) for b in self.config.bands}
 
         # pull the column indices
         idx_ra = cols.index(ra_label)
         idx_dec = cols.index(dec_label)
+        idx_redshift = cols.index(self.config.redshift_col)
         idx_n_obj = cols.index('n_obj')
         idx_brightest_flux = cols.index('brightest_flux')
         idx_total_flux = cols.index('total_flux')
@@ -139,14 +143,16 @@ class UnrecBlModel(Degrader):
 
             ## sum up the fluxes into the blended source
             for b in self.config.bands:
-                mergeData[i, cols.index(b)] = -2.5*np.log10(np.sum(these_fluxes[b]))
+                mergeData[i, cols.index(b)] = -2.5*np.log10(np.sum(these_fluxes[b])) + self.config.zp_dict[b]
 
             brighest_idx = np.argmax(ref_fluxes)
+            redshifts = these_redshifts.iloc[brighest_idx]
 
+            mergeData[i, idx_redshift] = redshifts
             mergeData[i, idx_n_obj] = n_obj
             mergeData[i, idx_brightest_flux] = ref_fluxes.max()
             mergeData[i, idx_total_flux] = np.sum(ref_fluxes)
-            mergeData[i, idx_z_brightest] = these_redshifts.iloc[brighest_idx]
+            mergeData[i, idx_z_brightest] = redshifts
             mergeData[i, idx_z_mean] = np.mean(these_redshifts)
             mergeData[i, idx_z_weighted] = np.sum(these_redshifts*ref_fluxes)/np.sum(ref_fluxes)
             if n_obj > 1:

rail/pipelines/degradation/apply_phot_errors.py

@@ -13,6 +13,7 @@ from rail.core.stage import RailStage, RailPipeline
 import ceci
 
 from rail.core.utils import RAILDIR
+from rail.utils import catalog_utils
 
 if 'PZ_DUSTMAP_DIR' not in os.environ:
     os.environ['PZ_DUSTMAP_DIR'] = '.'
@@ -62,6 +63,7 @@ class ApplyPhotErrorsPipeline(RailPipeline):
                 name=f'error_model_{key}',
                 connections=dict(input=previous_stage.io.output),
                 hdf5_groupname='',
+                renameDict=catalog_utils.CatalogConfigBase.active_class().band_name_dict(),
             )
             self.add_stage(the_error_model)
             previous_stage = the_error_model

rail/pipelines/degradation/truth_to_observed.py

@@ -13,7 +13,7 @@ from rail.core.stage import RailStage, RailPipeline
 import ceci
 
 from rail.core.utils import RAILDIR
-
+from rail.utils import catalog_utils
 from rail.creation.degraders.unrec_bl_model import UnrecBlModel
 
 from .spectroscopic_selection_pipeline import SELECTORS, CommonConfigParams
@@ -36,6 +36,9 @@ class TruthToObservedPipeline(RailPipeline):
         DS = RailStage.data_store
         DS.__class__.allow_overwrite = True
 
+        active_catalog_config = catalog_utils.CatalogConfigBase.active_class()
+        band_name_dict = active_catalog_config.band_name_dict()
+
         if error_models is None:
             error_models = ERROR_MODELS.copy()
 
@@ -43,6 +46,8 @@ class TruthToObservedPipeline(RailPipeline):
             selectors = SELECTORS.copy()
 
         config_pars = CommonConfigParams.copy()
+        config_pars['colnames'] = band_name_dict.copy()
+        config_pars['colnames']['redshift'] = active_catalog_config.redshift_col
 
         self.reddener = Reddener.build(
             dustmap_dir=dustmap_dir,
@@ -60,6 +65,7 @@ class TruthToObservedPipeline(RailPipeline):
                 name=f'error_model_{key}',
                 connections=dict(input=previous_stage.io.output),
                 hdf5_groupname='',
+                renameDict=band_name_dict,
             )
             self.add_stage(the_error_model)
             previous_stage = the_error_model