update: file names

CITCOM-project · Dec 10, 2024 · 802e274 · 802e274
1 parent 029fdfc
commit 802e274
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git a/dafni/tests/__init__.py b/dafni/tests/__init__.py
diff --git a/dafni/tests/tests_main_dafni.py → dafni/tests/test_main_dafni.py b/dafni/tests/tests_main_dafni.py → dafni/tests/test_main_dafni.py
@@ -6,13 +6,12 @@
 from causal_testing.estimation.linear_regression_estimator import LinearRegressionEstimator
 from causal_testing.specification.causal_dag import CausalDAG
 from causal_testing.specification.variable import Input, Output
-from main_dafni import get_args, parse_variables
+from dafni.src.main_dafni import get_args, parse_variables
 
 # Base directory (relative to the current test file location)
 BASE_DIR = Path(__file__).resolve().parent.parent  # Points to ./dafni
 DATA_DIR = BASE_DIR / "data"  # Points to ./dafni/data
 
-
 class TestGetArgs(unittest.TestCase):
     """Test the argparse functionality of the DAFNI entrypoint."""