Apply suggestions from code review

Co-authored-by: coderabbitai[bot] <136622811+coderabbitai[bot]@users.noreply.github.com> Signed-off-by: Jinzhe Zeng <[email protected]>
deepmodeling · May 24, 2024 · fa8b2a6 · fa8b2a6
1 parent c7d560f
commit fa8b2a6
Show file tree

Hide file tree

Showing 3 changed files with 0 additions and 13 deletions.
diff --git a/source/tests/tf/test_descrpt_se_a_type.py b/source/tests/tf/test_descrpt_se_a_type.py
@@ -197,7 +197,6 @@ def test_descriptor_one_side(self):
         test_size = jdata["numb_test"]
         batch_size = 1
         test_size = 1
-        stop_batch = jdata["stop_batch"]
         rcut = jdata["model"]["descriptor"]["rcut"]
         sel = jdata["model"]["descriptor"]["sel"]
         ntypes = len(sel)

diff --git a/source/tests/tf/test_model_se_atten.py b/source/tests/tf/test_model_se_atten.py
@@ -49,7 +49,6 @@ def test_model(self):
 
         systems = jdata["systems"]
         set_pfx = "set"
-        batch_size = jdata["batch_size"]
         test_size = jdata["numb_test"]
         batch_size = 1
         test_size = 1
@@ -202,8 +201,6 @@ def test_exclude_types(self):
 
         systems = jdata["systems"]
         set_pfx = "set"
-        batch_size = jdata["batch_size"]
-        test_size = jdata["numb_test"]
         batch_size = 1
         test_size = 1
         rcut = jdata["model"]["descriptor"]["rcut"]
@@ -275,8 +272,6 @@ def test_compressible_model(self):
 
         systems = jdata["systems"]
         set_pfx = "set"
-        batch_size = jdata["batch_size"]
-        test_size = jdata["numb_test"]
         batch_size = 1
         test_size = 1
         rcut = jdata["model"]["descriptor"]["rcut"]
@@ -431,7 +426,6 @@ def test_compressible_exclude_types(self):
         systems = jdata["systems"]
         set_pfx = "set"
         batch_size = jdata["batch_size"]
-        test_size = jdata["numb_test"]
         batch_size = 1
         test_size = 1
         rcut = jdata["model"]["descriptor"]["rcut"]
@@ -505,11 +499,9 @@ def test_stripped_type_embedding_model(self):
 
         systems = jdata["systems"]
         set_pfx = "set"
-        batch_size = jdata["batch_size"]
         test_size = jdata["numb_test"]
         batch_size = 1
         test_size = 1
-        stop_batch = jdata["stop_batch"]
         rcut = jdata["model"]["descriptor"]["rcut"]
 
         data = DataSystem(systems, set_pfx, batch_size, test_size, rcut, run_opt=None)
@@ -744,11 +736,8 @@ def test_smoothness_of_stripped_type_embedding_smooth_model(self):
 
         systems = jdata["systems"]
         set_pfx = "set"
-        batch_size = jdata["batch_size"]
-        test_size = jdata["numb_test"]
         batch_size = 1
         test_size = 1
-        stop_batch = jdata["stop_batch"]
         rcut = jdata["model"]["descriptor"]["rcut"]
 
         data = DataSystem(systems, set_pfx, batch_size, test_size, rcut, run_opt=None)

diff --git a/source/tests/tf/test_polar_se_a_tebd.py b/source/tests/tf/test_polar_se_a_tebd.py
@@ -50,7 +50,6 @@ def test_model(self):
         test_size = jdata["numb_test"]
         batch_size = 1
         test_size = 1
-        stop_batch = jdata["stop_batch"]
         rcut = jdata["model"]["descriptor"]["rcut"]
 
         data = DataSystem(systems, set_pfx, batch_size, test_size, rcut, run_opt=None)