.

Author: smilesun

causalspyne/dag_viewer.py

@@ -57,6 +57,10 @@ def __init__(self, dag, rng=default_rng(0)):
         self._list_nodes2hide = None
         self._success = False
 
+    @property
+    def dag(self):
+        return self._dag
+
     def run(self, num_samples, list_nodes2hide=None, confound=False):
         """
         generate subgraph adjcency matrix and corresponding data
@@ -173,6 +177,14 @@ def to_csv(self, title="data_subdag.csv"):
     def list_global_inds_nodes2hide(self):
         return self._list_global_inds_unobserved
 
+    @property
+    def col_inds(self):
+        subview_global_inds = \
+            [self.dag._dict_node_names2ind[name]
+             for name in self.dag.list_node_names
+             if name not in self.str_node2hide]
+        return subview_global_inds
+
     @property
     def list_global_inds_observed(self):
         if self._list_global_inds_observed is None:

causalspyne/main.py

@@ -116,8 +116,4 @@ def re_hide(subview, dag, num_sample, list_confounder2hide, output_dir,
                 ",".join(str(node) for node in
                          subview._list_global_inds_unobserved)
             )
-    subview_global_inds = [dag._dict_node_names2ind[name]
-                           for name in dag.list_node_names if
-                           name not in str_node2hide]
-
-    return subview, dag, subview_global_inds
+    return subview

tests/test_utils_ancestral_acc.py

@@ -11,7 +11,7 @@ def test_ancestral_acc():
     """
     test if ancestral accuracy runs
     """
-    arr_data, dag, subview_global_inds = gen_partially_observed(
+    subview = gen_partially_observed(
         size_micro_node_dag=3,
         num_macro_nodes=2,
         degree=2,  # average vertex/node degree
@@ -23,8 +23,10 @@ def test_ancestral_acc():
         rng=np.random.default_rng(1),
         graphviz=False
     )
-    pred_order_inds = [dag._dict_node_names2ind[name] for name in dag.list_node_names]
-    acc = ancestral_acc(true_dag=dag,
+    pred_order_inds = \
+        [subview.dag._dict_node_names2ind[name]
+         for name in subview.dag.list_node_names]
+    acc = ancestral_acc(true_dag=subview.dag,
                         pred_order=pred_order_inds)
 
     print(f"ancestral acc: {acc}")