Convert to generators

Author: christopher-wild

causal_testing/specification/causal_dag.py

@@ -150,19 +150,19 @@ def check_iv_assumptions(self, treatment, outcome, instrument) -> bool:
             raise ValueError(f"Instrument {instrument} is not associated with treatment {treatment} in the DAG")
 
         # (ii) Instrument does not affect outcome except through its potential effect on treatment
-        if not all([treatment in path for path in nx.all_simple_paths(self.graph, source=instrument, target=outcome)]):
+        if not all((treatment in path for path in nx.all_simple_paths(self.graph, source=instrument, target=outcome))):
             raise ValueError(
                 f"Instrument {instrument} affects the outcome {outcome} other than through the treatment {treatment}"
             )
 
         # (iii) Instrument and outcome do not share causes
         if any(
-            [
+            (
                 cause
                 for cause in self.graph.nodes
                 if list(nx.all_simple_paths(self.graph, source=cause, target=instrument))
                 and list(nx.all_simple_paths(self.graph, source=cause, target=outcome))
-            ]
+            )
         ):
             raise ValueError(f"Instrument {instrument} and outcome {outcome} share common causes")