fix distributions bug

Author: christopher-wild

causal_testing/json_front/json_class.py

@@ -66,18 +66,16 @@ def set_path(self, json_path: str, dag_path: str, data_path: str):
         self.dag_path = Path(dag_path)
         self.data_path = Path(data_path)
 
-    def set_variables(self, inputs: dict, outputs: dict, metas: dict, distributions: dict, populates: dict):
+    def set_variables(self, inputs: dict, outputs: dict, metas: dict):
         """ Populate the Causal Variables
             :param inputs:
             :param outputs:
             :param metas:
-            :param distributions:
-            :param populates:
         """
         self.inputs = [Input(i['name'], i['type'], i['distribution']) for i in
                        inputs]
         self.outputs = [Output(i['name'], i['type']) for i in outputs]
-        self.metas = [Meta(i['name'], i['type'], [i['populate']]) for i in
+        self.metas = [Meta(i['name'], i['type'], i['populate']) for i in
                       metas] if metas else list()
 
     def setup(self):

examples/poisson/run_causal_tests.py

@@ -164,7 +164,7 @@ def add_modelling_assumptions(self, estimation_model: Estimator):
                           args.data_path)  # Set the path to the data.csv, dag.dot and causal_tests.json file
 
     # Load the Causal Variables into the JsonUtility class ready to be used in the tests
-    json_utility.set_variables(inputs, outputs, metas, distributions, populates)
+    json_utility.set_variables(inputs, outputs, metas)
     json_utility.setup()  # Sets up all the necessary parts of the json_class needed to execute tests
 
     json_utility.execute_tests(effects, mutates, estimators, args.f)