From 34befcf88311db218c067f2cc11aa8c6402d2a6b Mon Sep 17 00:00:00 2001
From: "pre-commit-ci[bot]"
 <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Date: Thu, 29 Feb 2024 11:37:11 +0000
Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
---
 pvnet_summation/models/base_model.py | 15 +++++++++------
 requirements.txt                     |  2 +-
 2 files changed, 10 insertions(+), 7 deletions(-)

diff --git a/pvnet_summation/models/base_model.py b/pvnet_summation/models/base_model.py
index aa3b488..e226112 100644
--- 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
index 46d04d2..793db32 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -18,4 +18,4 @@ tensorboard
 tqdm
 omegaconf
 hydra-core
-python-dotenv
\ No newline at end of file
+python-dotenv