fix minor bug related to the case in which dataset is set to None, in…

… class 'StatCondIndep'.
IntelLabs · May 2, 2023 · b7e6f7b · b7e6f7b
1 parent 299ac4a
commit b7e6f7b
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/causal_discovery_utils/cond_indep_tests.py b/causal_discovery_utils/cond_indep_tests.py
@@ -165,7 +165,9 @@ def __init__(self,
         else:
             self.retained_graph = retained_edges
 
-        node_size = get_var_size(data)
+        node_size = None
+        if data is not None:
+            node_size = get_var_size(data)
 
         self.data = data
         self.num_records = num_records

diff --git a/notebooks/causal_discovery_with_a_perfect_oracle.ipynb b/notebooks/causal_discovery_with_a_perfect_oracle.ipynb
@@ -221,4 +221,4 @@
  },
  "nbformat": 4,
  "nbformat_minor": 1
-}
+}
-Original file line number
+Diff line change
@@ Expand Up / @@ -221,4 +221,4 @@ @@
      },
      "nbformat": 4,
      "nbformat_minor": 1
-    }
+    }