Merge branch 'devel' of https://github.com/cherryWangY/deepmd-kit int…

…o devel
deepmodeling · Oct 26, 2024 · 619e5fb · 619e5fb
2 parents 4d3c1b8 + b92c1f8
commit 619e5fb
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 8 deletions.
diff --git a/source/tests/pt/model/test_compressed_descriptor_se_a.py b/source/tests/pt/model/test_compressed_descriptor_se_a.py
@@ -45,10 +45,7 @@ def eval_pt_descriptor(
     return result
 
 
-@parameterized(
-    ("float32", "float64"),
-    (True, False)
-)
+@parameterized(("float32", "float64"), (True, False))
 class TestDescriptorSeA(unittest.TestCase):
     def setUp(self):
         (self.dtype, self.type_one_side) = self.param

diff --git a/source/tests/pt/model/test_compressed_descriptor_se_atten.py b/source/tests/pt/model/test_compressed_descriptor_se_atten.py
@@ -45,10 +45,7 @@ def eval_pt_descriptor(
     return result
 
 
-@parameterized(
-    ("float32", "float64"),
-    (True, False)
-)
+@parameterized(("float32", "float64"), (True, False))
 class TestDescriptorSeAtten(unittest.TestCase):
     def setUp(self):
         (self.dtype, self.type_one_side) = self.param