Zero-out all data

amarquand · Aug 16, 2023 · a13f907 · a13f907
1 parent 4fe979e
commit a13f907
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/pcntoolkit/model/hbr.py b/pcntoolkit/model/hbr.py
@@ -392,8 +392,13 @@ def estimate(self, X, y, batch_effects, **kwargs):
                     dummy_array = xarray.DataArray(data = np.zeros((len(chain), len(draw), 1)), coords = {'chain':chain, 'draw':draw,'empty':np.array([0])}, name=j)
                     self.idata.posterior[j] = dummy_array
                     self.vars_to_sample.append(j)
-            self.idata.observed_data = None
-            self.idata.constant_data = None
+
+            # zero-out all data
+            for i in self.idata.constant_data.data_vars:
+                self.idata.constant_data[i] *= 0
+            for i in self.idata.observed_data.data_vars:
+                self.idata.observed_data[i] *= 0
+
         return self.idata
 
     def predict(