fix(tensorstore): Linting

openclimatefix · Nov 4, 2024 · d836b23 · d836b23
1 parent 1b17f0f
commit d836b23
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/src/nwp_consumer/internal/entities/tensorstore.py b/src/nwp_consumer/internal/entities/tensorstore.py
@@ -143,7 +143,7 @@ def initialize_empty_store(
             store_range = f"{coords.init_time[0]:%Y%m%d%H}-{coords.init_time[-1]:%Y%m%d%H}"
 
         store_path = pathlib.Path(
-            os.getenv("ZARRDIR", f"~/.local/cache/nwp/{repository}/{model}/data")
+            os.getenv("ZARRDIR", f"~/.local/cache/nwp/{repository}/{model}/data"),
         ) / f"{store_range}.zarr"
 
         # * Define a set of chunks allowing for intermediate parallel writes

diff --git a/src/nwp_consumer/internal/repositories/model_repositories/test_noaa_gfs.py b/src/nwp_consumer/internal/repositories/model_repositories/test_noaa_gfs.py
@@ -22,7 +22,7 @@ class TestECMWFRealTimeS3ModelRepository(unittest.TestCase):
     @unittest.skipIf(
         condition="CI" in os.environ,
         reason="Skipping integration test that requires S3 access.",
-    ) # TODO: Move into integration tests, or remove
+    )  # TODO: Move into integration tests, or remove
     def test__download_and_convert(self) -> None:
         """Test the _download_and_convert method."""
 
@@ -47,7 +47,7 @@ def test__download_and_convert(self) -> None:
         ]
 
         for url in urls:
-            with (self.subTest(url=url)):
+            with self.subTest(url=url):
                 result = c._download_and_convert(url)
 
                 self.assertTrue(is_successful(result), msg=f"Error: {result}")
@@ -105,5 +105,6 @@ class TestCase:
                 result = NOAAGFSS3ModelRepository._wanted_file(
                     filename=t.filename,
                     it=test_it,
-                    max_step=max(NOAAGFSS3ModelRepository.model().expected_coordinates.step))
+                    max_step=max(NOAAGFSS3ModelRepository.model().expected_coordinates.step),
+                )
                 self.assertEqual(result, t.expected)