From 9db60300f4879ac98fcb0ce0fa4dde09a3cb4ab7 Mon Sep 17 00:00:00 2001 From: brianreicher Date: Mon, 4 Dec 2023 18:38:59 -0500 Subject: [PATCH] Linting models --- tests/models/aclsd_test.py | 3 +-- tests/models/mtlsd_test.py | 1 - tests/models/stelarr_test.py | 1 - 3 files changed, 1 insertion(+), 4 deletions(-) diff --git a/tests/models/aclsd_test.py b/tests/models/aclsd_test.py index 46cb86f..5479132 100644 --- a/tests/models/aclsd_test.py +++ b/tests/models/aclsd_test.py @@ -6,7 +6,6 @@ class TestACLSDModel(unittest.TestCase): - def setUp(self): # Set up any required data or configuration for your tests unet = setup_unet( @@ -14,7 +13,7 @@ def setUp(self): 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 index b58454b..32566b8 100644 --- 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 index aac0bfa..af2a481 100644 --- 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)