test: capture log by explicitly setting logger name deepmd

Author: caic99

source/tests/pt/test_dploaderset.py

@@ -72,10 +72,9 @@ def test_batchsize(self) -> None:
         # test: batch_size:str = "filter"
         self.assertEqual(self.get_batch_sizes("filter:193"), 1)
         self.assertEqual(self.get_batch_sizes("filter:192"), 1)
-        with self.assertLogs() as cm:
+        with self.assertLogs(logger="deepmd") as cm:
             self.assertRaises(AssertionError, self.get_batch_sizes, "filter:191")
-            print(cm.output)  # DEBUG
-            self.assertIn("Remove 1 systems with more than 191 atoms", cm.output[-1])
+        self.assertIn("Remove 1 systems with more than 191 atoms", cm.output[-1])
 
         # test: unknown batch_size: str
         with self.assertRaises(ValueError) as context: