fix imports

Signed-off-by: Jinzhe Zeng <[email protected]>
deepmodeling · Sep 10, 2024 · 2b66476 · 2b66476
1 parent 0a7f598
commit 2b66476
Showing 1 changed file with 1 addition and 5 deletions.
diff --git a/tests/entrypoint/test_submit.py b/tests/entrypoint/test_submit.py
@@ -6,12 +6,8 @@
     Path,
 )
 
-try:
-    from context import dpgen2  # noqa: F401
-except ModuleNotFoundError:
-    # case of upload everything to argo, no context needed
-    pass
 # isort: off
+from .context import dpgen2  # noqa: F401
 from dpgen2.entrypoint.submit import (
     copy_scheduler_plans,
     expand_idx,