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

for more information, see https://pre-commit.ci
openclimatefix · Feb 29, 2024 · 34befcf · 34befcf
1 parent b650c2d
commit 34befcf
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 7 deletions.
diff --git a/pvnet_summation/models/base_model.py b/pvnet_summation/models/base_model.py
@@ -74,19 +74,22 @@ def __init__(
         self._accumulated_y_hat = PredAccumulator()
         self._accumulated_y_sum = PredAccumulator()
         self._accumulated_times = PredAccumulator()
-        
+
         self.use_quantile_regression = self.output_quantiles is not None
-        
+
         if self.use_quantile_regression:
             self.num_output_features = self.forecast_len_30 * len(self.output_quantiles)
         else:
             self.num_output_features = self.forecast_len_30
-
-
+
         if self.pvnet_model.use_quantile_regression:
-            self.pvnet_output_shape= (317, self.pvnet_model.forecast_len, len(self.pvnet_model.output_quantiles))
+            self.pvnet_output_shape = (
+                317,
+                self.pvnet_model.forecast_len,
+                len(self.pvnet_model.output_quantiles),
+            )
         else:
-            self.pvnet_output_shape= (317, self.pvnet_model.forecast_len)
+            self.pvnet_output_shape = (317, self.pvnet_model.forecast_len)
 
     def predict_pvnet_batch(self, batch):
         """Use PVNet model to create predictions for batch"""

diff --git a/requirements.txt b/requirements.txt
@@ -18,4 +18,4 @@ tensorboard
 tqdm
 omegaconf
 hydra-core
-python-dotenv
+python-dotenv