From 987aaf1e004aec0b25421b96ccf5c75192783889 Mon Sep 17 00:00:00 2001
From: Zhenyu Wang <wzy@calypso.cn>
Date: Sat, 30 Mar 2024 12:42:37 +0800
Subject: [PATCH] fix: fix ut

---
 tests/test_caly_evo_step.py     | 2 +-
 tests/test_dpgen_loop.py        | 4 ++--
 tests/test_prep_run_caly.py     | 2 +-
 tests/test_prep_run_dp_train.py | 2 +-
 tests/test_prep_run_lmp.py      | 2 +-
 tests/test_prep_run_vasp.py     | 2 +-
 tests/test_select_confs.py      | 2 +-
 7 files changed, 8 insertions(+), 8 deletions(-)

diff --git a/tests/test_caly_evo_step.py b/tests/test_caly_evo_step.py
index bc651e09..c3b03a5e 100644
--- a/tests/test_caly_evo_step.py
+++ b/tests/test_caly_evo_step.py
@@ -272,7 +272,7 @@ def tearDown(self):
             shutil.rmtree(i, ignore_errors=True)
         for i in Path().glob("caly_task*"):
             shutil.rmtree(i, ignore_errors=True)
-        shutil.rmtree("upload", ignore_errors=True)
+        # shutil.rmtree("upload", ignore_errors=True)
 
     @unittest.skip("only need to run test_01")
     def test_00(self):
diff --git a/tests/test_dpgen_loop.py b/tests/test_dpgen_loop.py
index 79128eae..e7abc75f 100644
--- a/tests/test_dpgen_loop.py
+++ b/tests/test_dpgen_loop.py
@@ -98,14 +98,14 @@
 )
 from dpgen2.utils.step_config import normalize as normalize_step_dict
 
-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,
     MockedCollectDataCheckOptParam,
     MockedCollectDataFailed,
diff --git a/tests/test_prep_run_caly.py b/tests/test_prep_run_caly.py
index f1e315ee..9a55dde8 100644
--- a/tests/test_prep_run_caly.py
+++ b/tests/test_prep_run_caly.py
@@ -80,7 +80,7 @@
 )
 from dpgen2.utils.step_config import normalize as normalize_step_dict
 
-from .mocked_ops import (
+from mocked_ops import (
     MockedCollRunCaly,
     MockedRunCalyModelDevi,
     MockedRunDPOptim,
diff --git a/tests/test_prep_run_dp_train.py b/tests/test_prep_run_dp_train.py
index 87ebe541..f28a9ee3 100644
--- a/tests/test_prep_run_dp_train.py
+++ b/tests/test_prep_run_dp_train.py
@@ -59,7 +59,7 @@
 )
 from dpgen2.utils.step_config import normalize as normalize_step_dict
 
-from .mocked_ops import (
+from mocked_ops import (
     MockedModifyTrainScript,
     MockedPrepDPTrain,
     MockedRunDPTrain,
diff --git a/tests/test_prep_run_lmp.py b/tests/test_prep_run_lmp.py
index 20ca570f..29e4fc0a 100644
--- a/tests/test_prep_run_lmp.py
+++ b/tests/test_prep_run_lmp.py
@@ -76,7 +76,7 @@
 )
 from dpgen2.utils.step_config import normalize as normalize_step_dict
 
-from .mocked_ops import (
+from mocked_ops import (
     MockedRunLmp,
     mocked_numb_models,
 )
diff --git a/tests/test_prep_run_vasp.py b/tests/test_prep_run_vasp.py
index ac306375..7f4bd5ca 100644
--- a/tests/test_prep_run_vasp.py
+++ b/tests/test_prep_run_vasp.py
@@ -65,7 +65,7 @@
 )
 from dpgen2.utils.step_config import normalize as normalize_step_dict
 
-from .mocked_ops import (
+from mocked_ops import (
     MockedPrepVasp,
     MockedRunVasp,
     mocked_incar_template,
diff --git a/tests/test_select_confs.py b/tests/test_select_confs.py
index 576ef36f..088e657c 100644
--- a/tests/test_select_confs.py
+++ b/tests/test_select_confs.py
@@ -49,7 +49,7 @@
     SelectConfs,
 )
 
-from .mocked_ops import (
+from mocked_ops import (
     MockedConfSelector,
     MockedExplorationReport,
     MockedSelectConfs,