fix: fix ut

wangzyphysics · Mar 29, 2024 · a760446 · a760446
1 parent d3dc01d
commit a760446
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/test_collect_data.py b/tests/test_collect_data.py
@@ -44,13 +44,14 @@
 except ModuleNotFoundError:
     # case of upload everything to argo, no context needed
     pass
-from .context import (
+from context import (
     default_host,
     default_image,
     skip_ut_with_dflow,
     skip_ut_with_dflow_reason,
     upload_python_packages,
 )
+
 from .mocked_ops import (
     MockedCollectData,
 )