merge

deepmodeling · Dec 4, 2024 · a119931 · a119931
2 parents 7b4259e + b697ac0
commit a119931
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 8 deletions.
diff --git a/tests/generator/test_make_fp.py b/tests/generator/test_make_fp.py
@@ -265,7 +265,7 @@ def _make_fake_md(idx, md_descript, atom_types, type_map, ele_temp=None):
             task_dir = os.path.join(
                 "iter.%06d" % idx,  # noqa: UP031
                 "01.model_devi",
-                "task.%03d.%06d" % (sidx, midx),  # noqa: UP031  # noqa: UP031
+                "task.%03d.%06d" % (sidx, midx),  # noqa: UP031
             )
             os.makedirs(os.path.join(task_dir, "traj"), exist_ok=True)
             for ii in range(nframes):
@@ -306,7 +306,7 @@ def _make_fake_md_merge_traj(idx, md_descript, atom_types, type_map, ele_temp=No
             task_dir = os.path.join(
                 "iter.%06d" % idx,  # noqa: UP031
                 "01.model_devi",
-                "task.%03d.%06d" % (sidx, midx),  # noqa: UP031  # noqa: UP031
+                "task.%03d.%06d" % (sidx, midx),  # noqa: UP031
             )
             cwd = os.getcwd()
             os.makedirs(task_dir, exist_ok=True)
@@ -851,7 +851,7 @@ def test_make_fp_amber_diff(self):
         task_dir = os.path.join(
             "iter.%06d" % 0,  # noqa: UP031
             "01.model_devi",
-            "task.%03d.%06d" % (0, 0),  # noqa: UP031  # noqa: UP031
+            "task.%03d.%06d" % (0, 0),  # noqa: UP031
         )
         os.makedirs(task_dir, exist_ok=True)
         with open(os.path.join(task_dir, "rc.mdout"), "w") as f:

diff --git a/tests/generator/test_make_md.py b/tests/generator/test_make_md.py
@@ -701,7 +701,7 @@ def test_restart_from_iter(self):
                 "iter.%06d" % 0,  # noqa: UP031
                 "01.model_devi",
                 "task.000.000000",
-                "rc.rst7",  # noqa: UP031
+                "rc.rst7",
             ),
             "w",
         ) as fw:
@@ -715,7 +715,7 @@ def test_restart_from_iter(self):
                 "iter.%06d" % 1,  # noqa: UP031
                 "01.model_devi",
                 "task.000.000000",
-                "init.rst7",  # noqa: UP031
+                "init.rst7",
             )
         ) as f:
             assert f.read() == restart_text

diff --git a/tests/generator/test_make_train.py b/tests/generator/test_make_train.py
@@ -206,7 +206,7 @@ def _make_fake_fp(iter_idx, sys_idx, nframes):
         dirname = os.path.join(
             "iter.%06d" % iter_idx,  # noqa: UP031
             "02.fp",
-            "task.%03d.%06d" % (sys_idx, ii),  # noqa: UP031  # noqa: UP031
+            "task.%03d.%06d" % (sys_idx, ii),  # noqa: UP031
         )
         os.makedirs(dirname, exist_ok=True)
     dirname = os.path.join("iter.%06d" % iter_idx, "02.fp", "data.%03d" % sys_idx)  # noqa: UP031  # noqa: UP031
@@ -549,7 +549,7 @@ def test_training_finetune_model(self):
                         "00.train",
                         "%03d" % ii,  # noqa: UP031
                         "old",
-                        "init.pb",  # noqa: UP031
+                        "init.pb",
                     )
                 )
             )
@@ -604,7 +604,7 @@ def test_training_init_frozen_model(self):
                         "00.train",
                         "%03d" % ii,  # noqa: UP031
                         "old",
-                        "init.pb",  # noqa: UP031
+                        "init.pb",
                     )
                 )
             )