lint

Author: dfulu

ocf_data_sampler/select/select_spatial_slice.py

@@ -61,7 +61,7 @@ def select_spatial_slice_pixels(
         The selected DataArray slice.
 
     Raises:
-        ValueError: If the window dimensions are not even or the slice extends beyond the data 
+        ValueError: If the window dimensions are not even or the slice extends beyond the data
             boundaries.
     """
     if (width_pixels % 2) != 0:
@@ -170,17 +170,17 @@ def select_spatial_slice_pixels_multiple(
             "Multi-location window not available: "
             f"left_idx ({left_idx}), right_idx ({right_idx}), "
             f"bottom_idx ({bottom_idx}), top_idx ({top_idx}), "
-            f"data_width_pixels ({data_width_pixels}), data_height_pixels ({data_height_pixels})"
+            f"data_width_pixels ({data_width_pixels}), data_height_pixels ({data_height_pixels})",
         )
-    
+
     # Add buffer of 1 pixel if window is 2 pixels wide to ensure the central location is within the
     # returned slice
     x_buffer = 1 if width_pixels==2 else 0
     y_buffer = 1 if height_pixels==2 else 0
 
     return da.isel(
         {
-            x_dim: slice(left_idx, right_idx+x_buffer), 
-            y_dim: slice(bottom_idx, top_idx+y_buffer)
-        }
+            x_dim: slice(left_idx, right_idx+x_buffer),
+            y_dim: slice(bottom_idx, top_idx+y_buffer),
+        },
     )

ocf_data_sampler/torch_datasets/utils/spatial_slice_for_dataset.py

@@ -1,12 +1,11 @@
 """Functions for selecting data around a given location."""
 
-import xarray as xr
 
 from ocf_data_sampler.config import Configuration
-from ocf_data_sampler.select.geospatial import find_coord_system
 from ocf_data_sampler.select.location import Location
 from ocf_data_sampler.select.select_spatial_slice import (
-    select_spatial_slice_pixels, select_spatial_slice_pixels_multiple
+    select_spatial_slice_pixels,
+    select_spatial_slice_pixels_multiple,
 )