Update hierarchical.py

y0-causal-inference · Oct 1, 2024 · c15c73b · c15c73b
1 parent 65e7b07
commit c15c73b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/y0/hierarchical.py b/src/y0/hierarchical.py
@@ -240,7 +240,7 @@ def augment_collapsed_model(
     if not mechanism <= collapsed.nodes():
         raise ValueError("The input mechanism must be contained in the collapsed model.")
     aug = augmentation_variable
-    augmented.add_node(aug) 
+    augmented.add_node(aug)
     for var in mechanism:
         augmented.add_directed_edge(var, aug)
     for var in set(augmented.nodes()) - {aug}: