Merge remote-tracking branch 'origin/jacob/more-tests' into jacob/more-tests

Author: jacobbieker

ocf_datapipes/validation/check_equality.py

@@ -13,6 +13,7 @@
 @functional_datapipe("check_greater_than_or_equal_to")
 class CheckGreaterThanOrEqualToIterDataPipe(IterDataPipe):
     """Check greater than or equal to"""
+
     def __init__(
         self, source_datapipe: IterDataPipe, min_value: int, dataset_name: Optional[str] = None
     ):
@@ -43,6 +44,7 @@ def __iter__(self) -> Union[xr.Dataset, xr.DataArray]:
 @functional_datapipe("check_less_than_or_equal_to")
 class CheckLessThanOrEqualToIterDataPipe(IterDataPipe):
     """Check less than or equal to equality"""
+
     def __init__(
         self, source_datapipe: IterDataPipe, max_value: int, dataset_name: Optional[str] = None
     ):
@@ -73,6 +75,7 @@ def __iter__(self) -> Union[xr.Dataset, xr.DataArray]:
 @functional_datapipe("check_not_equal_to")
 class CheckNotEqualToIterDataPipe(IterDataPipe):
     """Check not equal to equality"""
+
     def __init__(
         self,
         source_datapipe: IterDataPipe,

ocf_datapipes/validation/check_vars_and_dims.py

@@ -9,6 +9,7 @@
 @functional_datapipe("check_vars_and_dims")
 class CheckVarsAndDimsIterDataPipe(IterDataPipe):
     """Check that the variables and dims exist"""
+
     def __init__(
         self,
         source_datapipe: IterDataPipe,