Linting models

htem · Dec 4, 2023 · 9db6030 · 9db6030
1 parent 6916e8b
commit 9db6030
Show file tree

Hide file tree

Showing 3 changed files with 1 addition and 4 deletions.
diff --git a/tests/models/aclsd_test.py b/tests/models/aclsd_test.py
@@ -6,15 +6,14 @@
 
 
 class TestACLSDModel(unittest.TestCase):
-
     def setUp(self):
         # Set up any required data or configuration for your tests
         unet = setup_unet(
             in_channels=10,
             downsample_factors=[(2, 2, 2), (2, 2, 2)],
             padding="same",
             num_heads=1,
-        )        
+        )
         self.mtlsd_model = ACLSDModel(unet, unet.out_channels)
 
     def test_forward(self):

diff --git a/tests/models/mtlsd_test.py b/tests/models/mtlsd_test.py
@@ -6,7 +6,6 @@
 
 
 class TestMTLSDModel(unittest.TestCase):
-
     def setUp(self):
         # Set up any required data or configuration for your tests
         unet = setup_unet()

diff --git a/tests/models/stelarr_test.py b/tests/models/stelarr_test.py
@@ -6,7 +6,6 @@
 
 
 class TestSTELARRModel(unittest.TestCase):
-
     def setUp(self):
         # Set up any required data or configuration for your tests
         unet = setup_unet(downsample_factors=[(2, 2, 2), (2, 2, 2)], num_heads=3)