Merge pull request #39 from /issues/36

fixed TJPCov compatibility
LSSTDESC · Oct 17, 2023 · 91fc88a · 91fc88a
2 parents 08e1874 + ef57ce0
commit 91fc88a
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 13 deletions.
diff --git a/augur/generate.py b/augur/generate.py
@@ -321,16 +321,16 @@ def generate(config, return_all_outputs=False, write_sacc=True, force_read=True)
         ccl_tracers = dict()
         bias_all = dict()
         for i, myst1 in enumerate(lk.statistics):
-            trname1 = myst1.source0.sacc_tracer
-            trname2 = myst1.source1.sacc_tracer
-            tr1 = myst1.source0.tracers[0].ccl_tracer  # Pulling out the tracers
-            tr2 = myst1.source1.tracers[0].ccl_tracer
+            trname1 = myst1.statistic.source0.sacc_tracer
+            trname2 = myst1.statistic.source1.sacc_tracer
+            tr1 = myst1.statistic.source0.tracers[0].ccl_tracer  # Pulling out the tracers
+            tr2 = myst1.statistic.source1.tracers[0].ccl_tracer
             ccl_tracers[trname1] = tr1
             ccl_tracers[trname2] = tr2
             if 'lens' in trname1:
-                bias_all[trname1] = myst1.source0.bias
+                bias_all[trname1] = myst1.statistic.source0.bias
             if 'lens' in trname2:
-                bias_all[trname2] = myst1.source1.bias
+                bias_all[trname2] = myst1.statistic.source1.bias
         for key in bias_all.keys():
             tjpcov_config['tjpcov'][f'bias_{key}'] = bias_all[key]
         tjpcov_config['tjpcov']['sacc_file'] = S

diff --git a/examples/config_test.yml b/examples/config_test.yml
@@ -74,14 +74,14 @@ cov_options:
     # fsky : 0.3
     # ------
     # Or you can also get it from a file
-    cov_type : 'SRD'
-    SRD_cov_path : '../data/Y1_3x2_SRD_cov.npy'
+    # cov_type : 'SRD'
+    #SRD_cov_path : './data/Y1_3x2_SRD_cov.npy'
     # Or using TJPCov
-    # cov_type : 'tjpcov'
-    # IA : None
-    #fsky: 0.3
-    #binning_info : 
-    #    ell_edges : np.geomspace(20, 15000, 21, endpoint=True).astype(np.int32)
+    cov_type : 'tjpcov'
+    IA : 0.0
+    fsky: 0.3
+    binning_info : 
+        ell_edges : np.geomspace(20, 15000, 21, endpoint=True).astype(np.int32)
 
 fisher:
     parameters: