From 34ab9520f20977ab41ae08759f71e719902fcec2 Mon Sep 17 00:00:00 2001
From: Zhenyu Wang <wzy@calypso.cn>
Date: Sat, 30 Mar 2024 10:35:56 +0800
Subject: [PATCH] fix: fix ut

---
 tests/test_collect_data.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/tests/test_collect_data.py b/tests/test_collect_data.py
index be7d192d..95b7e4c9 100644
--- a/tests/test_collect_data.py
+++ b/tests/test_collect_data.py
@@ -44,14 +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 (
+from mocked_ops import (
     MockedCollectData,
 )