Update discover_latent_nodes.py

y0-causal-inference · Dec 14, 2023 · bf2ae23 · bf2ae23
1 parent c630187
commit bf2ae23
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/eliater/discover_latent_nodes.py b/src/eliater/discover_latent_nodes.py
@@ -171,8 +171,12 @@ def remove_nuisance_variables(
     :param tag: The tag for which variables are latent
     :return: the new graph after simplification
     """
+    rv = NxMixedGraph(
+        directed=graph.directed.copy(),
+        undirected=graph.undirected.copy(),
+    )
     lv_dag = mark_nuisance_variables_as_latent(
-        graph=graph, treatments=treatments, outcomes=outcomes, tag=tag
+        graph=rv, treatments=treatments, outcomes=outcomes, tag=tag
     )
     simplified_latent_dag = simplify_latent_dag(lv_dag, tag=tag)
     return NxMixedGraph.from_latent_variable_dag(simplified_latent_dag.graph, tag=tag)