fix typo

Signed-off-by: Jinzhe Zeng <[email protected]>
deepmodeling · Jan 26, 2024 · 908e0d7 · 908e0d7
1 parent 1c3452d
commit 908e0d7
Showing 1 changed file with 9 additions and 9 deletions.
diff --git a/source/tests/test_init_frz_model_se_atten.py b/source/tests/test_init_frz_model_se_atten.py
@@ -198,15 +198,15 @@ def tearDownClass(cls):
             _file_delete(cls.FROZEN_MODELS[i])
             _file_delete("out.json")
             _file_delete(str(tests_path / "checkpoint"))
-            _file_delete(cls.CKPT[i] + ".meta")
-            _file_delete(cls.CKPT[i] + ".index")
-            _file_delete(cls.CKPT[i] + ".data-00000-of-00001")
-            _file_delete(cls.CKPT[i] + "-0.meta")
-            _file_delete(cls.CKPT[i] + "-0.index")
-            _file_delete(cls.CKPT[i] + "-0.data-00000-of-00001")
-            _file_delete(cls.CKPT[i] + "-1.meta")
-            _file_delete(cls.CKPT[i] + "-1.index")
-            _file_delete(cls.CKPT[i] + "-1.data-00000-of-00001")
+            _file_delete(cls.CKPTS[i] + ".meta")
+            _file_delete(cls.CKPTS[i] + ".index")
+            _file_delete(cls.CKPTS[i] + ".data-00000-of-00001")
+            _file_delete(cls.CKPTS[i] + "-0.meta")
+            _file_delete(cls.CKPTS[i] + "-0.index")
+            _file_delete(cls.CKPTS[i] + "-0.data-00000-of-00001")
+            _file_delete(cls.CKPTS[i] + "-1.meta")
+            _file_delete(cls.CKPTS[i] + "-1.index")
+            _file_delete(cls.CKPTS[i] + "-1.data-00000-of-00001")
             _file_delete(f"input_v2_compat{i}.json")
             _file_delete("lcurve.out")