Selection effects (#19)

* Test data generator with magnitude selection

* Fix normalization

* Update plot

* Remove unneeded import

* Adjust plot range

* Fix testdata generation

* Add hubble diagramm

* Add Hubble-free selection effect model

* Add test data normalization option

* Add nominal m_cut parameter

* Initial refactor structure

* Refactor classic h0 free scm

* Refactor nh h0 free scm

* Refactor classic nh h0 free scm

* Refactor classic nh h0 scm

* Refactor classic nh h0 scm

* Refactor simple h0 scm

* Refactor simple h0 scm

* Refactor h0 scm

* Refactor classic h0 scm

* Add selection effect models to hubble scm

* Fix priors

* Update bootstrap for new model system

* Minor fixes for hubble scm

* Some fixes for hubbe scm test data

* Fix selection effect model for hubble scm

* Fix systematic uncertainties

* Add z_pv file support

* Add selection values to data collection

* Minor fix

* Remove selection effects model from calibs

* Update pv uncertainty

* Add option to not store chains

* Add interactive plot mode

* Minor fixes

* Fix units

* Remove TkAgg import

Author: AlexHls

.gitignore

@@ -3,3 +3,7 @@ sccala-dev
 build
 _build
 src/sccala_sniip.egg-info
+
+src/sccala/scmlib/models/*
+!src/sccala/scmlib/models/*.*
+

src/sccala/asynphot/aks_correction.py

@@ -41,9 +41,9 @@ def get_sn_phot(photometry_file):
     )
     data.sort_values(by=["mjd"])
     tels = data["telescope"].unique().tolist()
-    assert (
-        len(tels) == 1
-    ), "More than one telescope found. Please correct each telescope separately."
+    assert len(tels) == 1, (
+        "More than one telescope found. Please correct each telescope separately."
+    )
 
     phot_data = data.pivot_table(
         index=["mjd"], columns=["band"], values=["mag", "emag"]
@@ -267,10 +267,6 @@ def aks_correction(
     aks_corr_phot, aks_corr_phot_err
     """
 
-    try:
-        matplotlib.use("TkAgg")
-    except ImportError:
-        pass
     # SN data
     a_v, z_hel, mjd_explo = get_sn_info(snname, modelpath=modelpath)
 

src/sccala/collect_scm_data.py

@@ -25,6 +25,9 @@ def main(args):
     rho = args.rho
     rho_calib = args.rho_calib
     error_mode = args.error_mode
+    m_cut_nom = args.m_cut_nom
+    sig_cut_nom = args.sig_cut_nom
+    pv_red_file = args.pv_red_file
 
     df = sl.load_data(
         sne_list,
@@ -43,6 +46,9 @@ def main(args):
         rho=rho,
         rho_calib=rho_calib,
         error_mode=error_mode,
+        m_cut_nom=m_cut_nom,
+        sig_cut_nom=sig_cut_nom,
+        pv_red_file=pv_red_file,
     )
 
     return df
@@ -140,6 +146,21 @@ def cli():
         default="mean",
         choices=["mean", "max", "min"],
     )
+    parser.add_argument(
+        "--m_cut_nom",
+        help="Nominal value of the magnitude cut. Used for the full sample.",
+        default=18.5,
+    )
+    parser.add_argument(
+        "--sig_cut_nom",
+        help="Nominal value of the magnitude cut uncertainty. Used for the full sample.",
+        default=0.5,
+    )
+    parser.add_argument(
+        "-p",
+        "--pv_red_file",
+        help="Path to the file containing redshifts corrected for peculiar velocities.",
+    )
 
     args = parser.parse_args()