update config

Author: zaidikumail

scripts/config.yaml

@@ -15,3 +15,9 @@ epoch:
   feniks_n_centroids: 500
 
   num_halos: 300
+
+defaults:
+  diffstarpop: False
+  spspop: False
+  ssperr: False
+  merging: False

scripts/config_diagnostics.yaml

@@ -9,13 +9,13 @@ ssp_file: /Users/kumail/diffdir/ssp_data/ssp_w_emlines/fsps_v0.4.7_mist_c3k_a_kr
 
 plots:
   num_halos : 1000
-  plot_colors_mags: True
-  plot_mags: True
-  plot_ssperr: True
-  plot_massive_cen_colors: True
-  plot_satquench: True
-  plot_satquench_model: True
-  plot_insitu_smhm: True
-  plot_uvj: True
+  plot_colors_mags: False
+  plot_mags: False
+  plot_ssperr: False
+  plot_massive_cen_colors: False
+  plot_satquench: False
+  plot_satquench_model: False
+  plot_insitu_smhm: False
+  plot_uvj: False
   plot_exsitu_frac: True
-  plot_avpop: True
+  plot_avpop: False

scripts/fit_diffsky.py

@@ -8,6 +8,12 @@
 import numpy as np
 import yaml
 from diffsky.data_loaders.hacc_utils import lc_mock
+from diffsky.merging.merging_model import DEFAULT_MERGE_PARAMS
+from diffsky.param_utils.spspop_param_utils import DEFAULT_SPSPOP_PARAMS
+from diffsky.ssp_err_model.defaults import ZERO_SSPERR_PARAMS
+from diffstar.diffstarpop.kernels.params.params_diffstarpopfits_mgash import (
+    DiffstarPop_Params_Diffstarpopfits_mgash,
+)
 from dsps import load_ssp_templates
 from dsps.data_loaders import load_emline_info as lemi
 from jax import random as jran
@@ -18,6 +24,10 @@
 from diffhtwo.experimental.latin_hypercube import lh_utils as lhu
 from diffhtwo.experimental.optimizers import Np_specphot_opt
 
+DIFFSTARPOP_GALACTICUS_exsitu = DiffstarPop_Params_Diffstarpopfits_mgash[
+    "galacticus_in_plus_ex_situ"
+]
+
 if __name__ == "__main__":
     p = argparse.ArgumentParser()
     p.add_argument("--config", default="config.yaml")
@@ -52,6 +62,23 @@
         fit_start_drn,
         cfg["start_runid"] + "_" + cfg["start_fit_type"],
     )
+    if cfg["defaults"]["diffstarpop"]:
+        param_collection_fit = param_collection_fit._replace(
+            diffstarpop_params=DIFFSTARPOP_GALACTICUS_exsitu
+        )
+    if cfg["defaults"]["spspop"]:
+        param_collection_fit = param_collection_fit._replace(
+            spspop_params=DEFAULT_SPSPOP_PARAMS
+        )
+    if cfg["defaults"]["ssperr"]:
+        param_collection_fit = param_collection_fit._replace(
+            ssperr_params=ZERO_SSPERR_PARAMS
+        )
+    if cfg["defaults"]["merging"]:
+        param_collection_fit = param_collection_fit._replace(
+            merging_params=DEFAULT_MERGE_PARAMS
+        )
+
     u_theta_fit = pu.get_u_theta_from_param_collection(param_collection_fit)
 
     # fit dirs