Update source/tests/pt/test_dp_show.py

Co-authored-by: coderabbitai[bot] <136622811+coderabbitai[bot]@users.noreply.github.com>
Signed-off-by: Chenqqian Zhang <[email protected]>

source/tests/pt/test_dp_show.py

@@ -25,7 +25,7 @@
 class TestSingleTaskModel(unittest.TestCase):
     def setUp(self):
         input_json = str(Path(__file__).parent / "water/se_atten.json")
-        with open(input_json, "r") as f:
+        with open(input_json) as f:
             self.config = json.load(f)
         self.config["training"]["numb_steps"] = 1
         self.config["training"]["save_freq"] = 1
@@ -37,7 +37,6 @@ def setUp(self):
         trainer = get_trainer(deepcopy(self.config))
         trainer.run()
         os.system("dp --pt freeze")
-
     def test_checkpoint(self):
         INPUT = "model.pt"
         ATTRIBUTES = "type-map descriptor fitting-net"