change path method

Author: christopher-wild

causal_testing/json_front/json_class.py

@@ -46,18 +46,20 @@ def __init__(self):
         self.modelling_scenario = None
         self.causal_specification = None
 
-    def set_path(self, directory_path: Path):
+    def set_path(self, json_path: str, dag_path: str, data_path: str):
         """
             Takes a path of the directory containing all scenario specific files and creates individual paths for each file
-            :param directory_path: pathlib.Path pointing towards directory containing all scenario specific user code and files
+            :param json_path: string path representation to .json file containing test specifications
+            :param dag_path: string path representation to the .dot file containing the Causal DAG
+            :param data_path: string path representation to the data file
             :returns:
-                - json_path -  path to causal_tests.json
-                - dag_path -  path to dag.dot
-                - data_path - path to scenario data, expected in data.csv
+                - json_path -
+                - dag_path -
+                - data_path -
             """
-        self.json_path = directory_path.joinpath("causal_tests.json")
-        self.dag_path = directory_path.joinpath("dag.dot")
-        self.data_path = directory_path.joinpath("data.csv")
+        self.json_path = Path(json_path)
+        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):
         """ Populate the Causal Variables

examples/poisson/causal_test_setup.py

@@ -13,6 +13,11 @@
 from causal_testing.specification.scenario import Scenario
 from causal_testing.specification.variable import Input, Output, Meta
 
+data_path = "data.csv"
+dag_path = "dag.dot"
+json_path = "causal_tests.json"
+
+
 class WidthHeightEstimator(LinearRegressionEstimator):
     """
     Extension of LinearRegressionEstimator class to include scenario specific user code
@@ -104,8 +109,6 @@ def get_args() -> argparse.Namespace:
         description="A script for parsing json config files for the Causal Testing Framework")
     parser.add_argument("-f", help="if included, the script will stop if a test fails",
                         action="store_true")
-    parser.add_argument("--directory_path", help="path to the json file containing the causal tests",
-                        default="causal_tests.json", type=Path)
     return parser.parse_args()
 
 
@@ -188,7 +191,7 @@ def add_modelling_assumptions(self, estimation_model: Estimator):
     args = get_args()
 
     json_utility = MyJsonUtility()  # Create an instance of the extended JsonUtility class
-    json_utility.set_path(args.directory_path)  # Set the path to the data.csv, dag.dot and causal_tests.json file
+    json_utility.set_path(json_path, dag_path, 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)