Merge pull request #233 from jakobrunge/developer

further fixes for multiple dataset functionality
jakobrunge · Aug 4, 2022 · 6724e51 · 6724e51
2 parents b59f59a + b7d11b5
commit 6724e51
Show file tree

Hide file tree

Showing 3 changed files with 24 additions and 27 deletions.
diff --git a/setup.py b/setup.py
@@ -63,7 +63,7 @@ def run(self):
 # Run the setup
 setup(
     name="tigramite",
-    version="5.1.0.1",
+    version="5.1.0.2",
     packages=["tigramite", "tigramite.independence_tests", "tigramite.toymodels"],
     license="GNU General Public License v3.0",
     description="Tigramite causal discovery for time series",

diff --git a/tigramite/models.py b/tigramite/models.py
@@ -1230,7 +1230,7 @@ def __init__(self,
 
         # Default value for the mask
         if dataframe.mask is not None:
-            mask = dataframe.mask[0]
+            mask = {0: dataframe.mask[0]}
         else:
             mask = {0: np.zeros(dataframe.values[0].shape, dtype='bool')}
         # Get the dataframe shape