From a21404fadae333e438a013ce7009165871ae8d2a Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Mon, 24 Jul 2023 10:36:39 +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 | 2 -- pvnet_summation/models/model.py | 2 +- pvnet_summation/utils.py | 2 +- 3 files changed, 2 insertions(+), 4 deletions(-) diff --git a/pvnet_summation/models/base_model.py b/pvnet_summation/models/base_model.py index 93ff2e2..6db2eb3 100644 --- a/pvnet_summation/models/base_model.py +++ b/pvnet_summation/models/base_model.py @@ -16,7 +16,6 @@ from pvnet_summation.utils import plot_forecasts - logger = logging.getLogger(__name__) activities = [torch.profiler.ProfilerActivity.CPU] @@ -52,7 +51,6 @@ def __init__( ) self.pvnet_model.requires_grad_(False) - self._optimizer = optimizer # Model must have lr to allow tuning diff --git a/pvnet_summation/models/model.py b/pvnet_summation/models/model.py index 47132f1..30e9486 100644 --- a/pvnet_summation/models/model.py +++ b/pvnet_summation/models/model.py @@ -104,4 +104,4 @@ def forward(self, x): out = F.leaky_relu(gsp_sum + out) - return out \ No newline at end of file + return out diff --git a/pvnet_summation/utils.py b/pvnet_summation/utils.py index 91214de..9f7d179 100644 --- a/pvnet_summation/utils.py +++ b/pvnet_summation/utils.py @@ -24,7 +24,7 @@ def plot_forecasts(y, y_hat, times, batch_idx=None, quantiles=None, y_sum=None): continue ax.plot(times_utc[i], y[i], marker=".", color="k", label=r"$y$") - + if y_sum is not None: ax.plot( times_utc[i], y_sum[i], marker=".", linestyle="--", color="k", label=r"$y_{sum}$"