fixing black error

UrbanSystemsLab · Nov 27, 2024 · d7ec502 · d7ec502
1 parent a1e1eb4
commit d7ec502
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/usl_models/tests/atmo_ml/dataset_test.py b/usl_models/tests/atmo_ml/dataset_test.py
@@ -92,13 +92,15 @@ def test_load_dataset_structure(self, mock_storage_client):
                     "spatial": (B, H, W, F_S),
                     "lu_index": (B, H, W),
                 }
-            ] * num_batches,
+            ]
+            * num_batches,
         )
         self.assertShapesRecursive(
             list(labels),
             [
                 (B, T_O, H, W, C),
-            ] * num_batches,
+            ]
+            * num_batches,
         )
 
     @mock.patch("google.cloud.storage.Client")

diff --git a/usl_models/usl_models/testing/__init__.py b/usl_models/usl_models/testing/__init__.py
@@ -7,6 +7,7 @@
 
 class TestCase(unittest.TestCase):
     """Testing utils."""
+
     def assertShapesRecursive(self, obj: object, expected: object, path: str = ""):
         """Recursively checks the shapes of numpy arrays in a data structure."""
         if isinstance(obj, np.ndarray) or tf.is_tensor(obj):