add explicit path to load_statuses.py

mlibrary · Sep 24, 2024 · 3f88dc1 · 3f88dc1
1 parent d5aab0f
commit 3f88dc1
Showing 1 changed file with 9 additions and 5 deletions.
diff --git a/aim/digifeeds/bin/load_statuses.py b/aim/digifeeds/bin/load_statuses.py
@@ -1,10 +1,14 @@
-from aim.digifeeds.database.models import load_statuses
-from aim.digifeeds.database.main import SessionLocal
 import sys
 
+sys.path = ["", "../../"] + sys.path[1:]
+from aim.digifeeds.database.models import load_statuses  # noqa: E402
+from aim.digifeeds.database.main import SessionLocal  # noqa: E402
+
+
 def main():
     with SessionLocal() as db_session:
-      load_statuses(session=db_session)
+        load_statuses(session=db_session)
+
 
-if __name__=='__main__':
-    sys.exit(main())
+if __name__ == "__main__":
+    sys.exit(main())