fix import paths

rewiringamerica · Nov 19, 2024 · 38c5cf9 · 38c5cf9
1 parent 25b0e82
commit 38c5cf9
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/tests/test_constants.py b/tests/test_constants.py
@@ -4,7 +4,7 @@
 import sys
 
 if os.environ.get("DATABRICKS_RUNTIME_VERSION", None):
-    sys.path.append("../src")
+    sys.path.append("src")
 
 import unittest
 from dmutils import constants

diff --git a/tests/test_utils.py b/tests/test_utils.py
@@ -8,9 +8,9 @@
 import unittest
 
 if os.environ.get("DATABRICKS_RUNTIME_VERSION", None):
-    sys.path.append("../src")
+    sys.path.append("src")
 
-from dmutils import constants, bsb, data_cleaning
+from dmutils import constants
 
 # TODO: Remove skips for unit test once we have spark testing working on git:
 # https://www.notion.so/rewiringamerica/Local-Spark-Testing-4aef885e20034c18b1a2fba6c355e82c?pvs=4