fix lint

sail-sg · Oct 27, 2023 · ea81714 · ea81714
1 parent f09bbfa
commit ea81714
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 3 deletions.
diff --git a/tests/integral/pw/pw_test.py b/tests/integral/pw/pw_test.py
@@ -25,7 +25,9 @@
 
 class PWTest(parameterized.TestCase):
 
-  @parameterized.parameters(("C2", 3.5667, 76.5484253352856),)
+  @parameterized.parameters(
+    ("C2", 3.5667, 76.5484253352856),
+  )
   def test_init_from_name_and_lattice(
     self, crystal_name: str, a: float, expected_vol: float
   ):

diff --git a/tests/native/xla/example_test.py b/tests/native/xla/example_test.py
@@ -7,6 +7,7 @@
 import numpy as np
 from absl import logging
 from absl.testing import absltest
+
 from d4ft.native.xla.custom_call import CustomCallMeta
 from tests.native.xla.example import Parent, _Example, _ExampleMember
 

diff --git a/tests/system/crystal_test.py b/tests/system/crystal_test.py
@@ -21,7 +21,9 @@
 
 class CrystalTest(parameterized.TestCase):
 
-  @parameterized.parameters(("C2", 3.5667, 76.5484253352856),)
+  @parameterized.parameters(
+    ("C2", 3.5667, 76.5484253352856),
+  )
   def test_init_from_name_and_lattice(
     self, crystal_name: str, a: float, expected_vol: float
   ):
@@ -39,7 +41,9 @@ def test_init_from_name_and_lattice(
 
     self.assertAlmostEqual(crystal.vol, expected_vol)
 
-  @parameterized.parameters(("diamond.xyz", 76.5484253352856),)
+  @parameterized.parameters(
+    ("diamond.xyz", 76.5484253352856),
+  )
   def test_init_from_xyz_file(self, xyz_file: str, expected_vol: float):
     crystal = Crystal.from_xyz_file(xyz_file)
     logging.info(f'cell volume:{crystal.vol}')