Merge branch 'tickets/DM-35807'

lsst · Aug 3, 2022 · 3d1ae88 · 3d1ae88
2 parents 296245c + 9a47956
commit 3d1ae88
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/python/lsst/cp/pipe/ptc/cpSolvePtcTask.py b/python/lsst/cp/pipe/ptc/cpSolvePtcTask.py
@@ -389,7 +389,7 @@ def fitDataFullCovariance(self, dataset):
                 dataset.covariancesModelNoB[ampName] = listNanMatrix
                 dataset.aMatrixNoB[ampName] = nanMatrix
 
-                dataset.expIdMask[ampName] = np.repeat(np.nan, lenInputTimes)
+                dataset.expIdMask[ampName] = np.repeat(False, lenInputTimes)
                 dataset.gain[ampName] = np.nan
                 dataset.gainErr[ampName] = np.nan
                 dataset.noise[ampName] = np.nan
@@ -440,8 +440,10 @@ def fitDataFullCovariance(self, dataset):
             dataset.ptcFitParsError[ampName] = [np.nan]
             dataset.ptcFitChiSq[ampName] = np.nan
 
-            # Save full covariances, covariances models, and their weights
-            # dataset.expIdMask is already full
+            # Save full covariances, covariances models, and their weights.
+            # dataset.expIdMask is already full, but needs to be
+            # converted to bool.
+            dataset.expIdMask[ampName] = np.array(dataset.expIdMask[ampName], dtype=bool)
             dataset.covariances[ampName] = covAtAmp
             dataset.covariancesModel[ampName] = self.evalCovModel(muAtAmp,
                                                                   fitResults['fullModel']['a'],