Skip to content

Commit 7c2f763

Browse files
update example_run_causal_tests.py
1 parent 275b32f commit 7c2f763

File tree

1 file changed

+3
-5
lines changed

1 file changed

+3
-5
lines changed

examples/poisson/example_run_causal_tests.py

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -135,7 +135,7 @@ def populate_num_shapes_unit(data):
135135
modelling_inputs = (
136136
[Input(i["name"], i["datatype"], i["distribution"]) for i in inputs]
137137
+ [Output(i["name"], i["datatype"]) for i in outputs]
138-
+ [Meta(i["name"], i["datatype"], [i["populate"]]) for i in metas]
138+
+ [Meta(i["name"], i["datatype"], i["populate"]) for i in metas]
139139
if metas
140140
else list()
141141
)
@@ -174,8 +174,7 @@ def test_run_causal_tests():
174174
) # Set the path to the data.csv, dag.dot and causal_tests.json file
175175

176176
# Load the Causal Variables into the JsonUtility class ready to be used in the tests
177-
json_utility.set_variables(inputs, outputs, metas)
178-
json_utility.setup() # Sets up all the necessary parts of the json_class needed to execute tests
177+
json_utility.setup(scenario=modelling_scenario) # Sets up all the necessary parts of the json_class needed to execute tests
179178

180179
json_utility.generate_tests(effects, mutates, estimators, False)
181180

@@ -188,7 +187,6 @@ def test_run_causal_tests():
188187
) # Set the path to the data.csv, dag.dot and causal_tests.json file
189188

190189
# Load the Causal Variables into the JsonUtility class ready to be used in the tests
191-
json_utility.set_variables(inputs, outputs, metas)
192-
json_utility.setup() # Sets up all the necessary parts of the json_class needed to execute tests
190+
json_utility.setup(scenario=modelling_scenario) # Sets up all the necessary parts of the json_class needed to execute tests
193191

194192
json_utility.generate_tests(effects, mutates, estimators, args.f)

0 commit comments

Comments
 (0)