[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

Author: pre-commit-ci[bot]

ocf_datapipes/convert/coordinates.py

@@ -1,7 +1,8 @@
+from typing import Union
+
+import xarray as xr
 from torchdata.datapipes import functional_datapipe
 from torchdata.datapipes.iter import IterDataPipe
-import xarray as xr
-from typing import Union
 
 from ocf_datapipes.utils.geospatial import (
     lat_lon_to_osgb,
@@ -13,6 +14,7 @@
 @functional_datapipe("convert_latlon_to_osgb")
 class ConvertLatLonToOSGBIterDataPipe(IterDataPipe):
     """Convert from Lat/Lon object to OSGB"""
+
     def __init__(self, source_datapipe: IterDataPipe):
         """
         Convert from Lat/Lon to OSGB
@@ -34,6 +36,7 @@ def __iter__(self) -> Union[xr.DataArray, xr.Dataset]:
 @functional_datapipe("convert_osgb_to_latlon")
 class ConvertOSGBToLatLonIterDataPipe(IterDataPipe):
     """Convert from OSGB to Lat/Lon"""
+
     def __init__(self, source_datapipe: IterDataPipe):
         """
         Convert from OSGB to Lat/Lon
@@ -55,6 +58,7 @@ def __iter__(self) -> Union[xr.DataArray, xr.Dataset]:
 @functional_datapipe("convert_geostationary_to_latlon")
 class ConvertGeostationaryToLatLonIterDataPipe(IterDataPipe):
     """Convert from geostationary to Lat/Lon points"""
+
     def __init__(self, source_datapipe: IterDataPipe):
         """
         Convert from Geostationary to Lat/Lon points and add to Xarray object

ocf_datapipes/convert/gsp.py

@@ -8,6 +8,7 @@
 @functional_datapipe("convert_gsp_to_numpy_batch")
 class ConvertGSPToNumpyBatchIterDataPipe(IterDataPipe):
     """Convert GSP Xarray to NumpyBatch"""
+
     def __init__(self, source_datapipe: IterDataPipe):
         """
         Convert GSP Xarray to NumpyBatch object

ocf_datapipes/convert/nwp.py

@@ -9,6 +9,7 @@
 @functional_datapipe("convert_nwp_to_numpy_batch")
 class ConvertNWPToNumpyBatchIterDataPipe(IterDataPipe):
     """Convert NWP Xarray objects to NumpyBatch ones"""
+
     def __init__(self, source_datapipe: IterDataPipe):
         """
         Convert NWP Xarray objecs to NumpyBatch ones

ocf_datapipes/convert/pv.py

@@ -9,6 +9,7 @@
 @functional_datapipe("convert_pv_to_numpy_batch")
 class ConvertPVToNumpyBatchIterDataPipe(IterDataPipe):
     """Convert PV Xarray to NumpyBatch"""
+
     def __init__(self, source_datapipe: IterDataPipe):
         """
         Convert PV Xarray objects to NumpyBatch objects

ocf_datapipes/convert/satellite.py

@@ -8,6 +8,7 @@
 @functional_datapipe("convert_satellite_to_numpy_batch")
 class ConvertSatelliteToNumpyBatchIterDataPipe(IterDataPipe):
     """Converts Xarray Satellite to NumpyBatch object"""
+
     def __init__(self, source_datapipe: IterDataPipe, is_hrv: bool = False):
         """
         Converts Xarray satellite object to NumpyBatch object

ocf_datapipes/select/location_picker.py

@@ -8,6 +8,7 @@
 @functional_datapipe("location_picker")
 class LocationPickerIterDataPipe(IterDataPipe):
     """Picks locations from a dataset and returns them"""
+
     def __init__(self, source_datapipe: IterDataPipe, return_all_locations: bool = False):
         """
         Picks locations from a dataset and returns them

tests/convert/test_gsp.py

@@ -4,6 +4,7 @@
 from ocf_datapipes.transform.xarray import AddT0IdxAndSamplePeriodDuration
 from ocf_datapipes.utils.consts import BatchKey
 
+
 def test_convert_gsp_to_numpy_batch(gsp_dp):
     gsp_dp = AddT0IdxAndSamplePeriodDuration(
         gsp_dp, sample_period_duration=timedelta(minutes=5), history_duration=timedelta(minutes=60)