Fix some linting

openclimatefix · Oct 17, 2023 · f71333c · f71333c
1 parent 1185bdb
commit f71333c
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/ocf_datapipes/training/pseudo_irradience.py b/ocf_datapipes/training/pseudo_irradience.py
@@ -418,6 +418,10 @@ def pseudo_irradiance_datapipe(
         batch_size: Batch size for the datapipe
         one_d: Whether to return a 1D array or not, i.e. a single PV site in the center as
             opposed to a 2D array of PV sites
+        size_meters: Size, in meters, of the output image
+        use_meters: Whether to use meters or pixels
+        normalize_by_pvlib: Whether to normalize the PV generation by the PVLib generation
+        is_test: Whether to return the test set or not
 
     Returns: datapipe
     """