Swap rng_seed with rng instance in treecorr config

lsst · Apr 19, 2024 · 00d0feb · 00d0feb
1 parent 85baccf
commit 00d0feb
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/python/lsst/analysis/tools/actions/vector/calcRhoStatistics.py b/python/lsst/analysis/tools/actions/vector/calcRhoStatistics.py
@@ -450,6 +450,10 @@ def __call__(self, data: KeyedData, **kwargs) -> KeyedData:
 
         treecorr_config_dict = self.treecorr.toDict()
 
+        # Swap rng_seed with an rng instance in treecorr config.
+        rng = np.random.RandomState(treecorr_config_dict.pop("rng_seed"))
+        treecorr_config_dict["rng"] = rng
+
         # Pass the appropriate arguments to the correlator and build a dict
         rhoStats: Mapping[str, treecorr.BinnedCorr2] = {}
         for rhoIndex in range(1, 6):