diff --git a/source/tests/tf/test_dipole_hybrid_descrpt.py b/source/tests/tf/test_dipole_hybrid_descrpt.py
index 7d49fc30b3..cc500c43ac 100644
--- 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"]