From 93e253eb04085a96d8451197a9b2314d34033f13 Mon Sep 17 00:00:00 2001 From: Duo <50307526+iProzd@users.noreply.github.com> Date: Wed, 6 Mar 2024 02:57:43 +0800 Subject: [PATCH] Update test_deeppot_a.py --- source/tests/tf/test_deeppot_a.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/source/tests/tf/test_deeppot_a.py b/source/tests/tf/test_deeppot_a.py index c12fd1063b..f40b57c213 100644 --- a/source/tests/tf/test_deeppot_a.py +++ b/source/tests/tf/test_deeppot_a.py @@ -804,7 +804,7 @@ def test_convert_012(self): convert_pbtxt_to_pb(str(infer_path / "sea_012.pbtxt"), old_model) run_dp(f"dp convert-from 0.12 -i {old_model} -o {new_model}") dp = DeepPot(new_model) - ret = dp.eval(self.coords, self.box, self.atype, atomic=True) + _ = dp.eval(self.coords, self.box, self.atype, atomic=True) os.remove(old_model) os.remove(new_model) @@ -814,7 +814,7 @@ def test_convert(self): convert_pbtxt_to_pb(str(infer_path / "sea_012.pbtxt"), old_model) run_dp(f"dp convert-from -i {old_model} -o {new_model}") dp = DeepPot(new_model) - ret = dp.eval(self.coords, self.box, self.atype, atomic=True) + _ = dp.eval(self.coords, self.box, self.atype, atomic=True) os.remove(old_model) os.remove(new_model)