update: file names

Author: f-allian

dafni/tests/__init__.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."""