Apply suggestions from code review

Signed-off-by: Jinzhe Zeng <[email protected]>
njzjz · Jan 8, 2025 · 088450e · 088450e
1 parent 6e01a3d
commit 088450e
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/source/tests/tf/test_dipole_hybrid_descrpt.py b/source/tests/tf/test_dipole_hybrid_descrpt.py
@@ -35,8 +35,6 @@ def test_model(self) -> None:
 
         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"]