From 25e033d8f74fd50a0caba06e31829c27d8e9f9c9 Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Fri, 19 Jan 2024 13:53:17 +0000 Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --- pvnet/models/multimodal/multimodal.py | 4 +--- pvnet/models/multimodal/site_encoders/encoders.py | 1 + 2 files changed, 2 insertions(+), 3 deletions(-) diff --git a/pvnet/models/multimodal/multimodal.py b/pvnet/models/multimodal/multimodal.py index 4cbbfb8d..c17cccfe 100644 --- a/pvnet/models/multimodal/multimodal.py +++ b/pvnet/models/multimodal/multimodal.py @@ -196,9 +196,7 @@ def __init__( if wind_history_minutes is None: wind_history_minutes = history_minutes - self.wind_encoder = wind_encoder( - sequence_length=self.history_len_30 - ) + self.wind_encoder = wind_encoder(sequence_length=self.history_len_30) # Update num features fusion_input_features += self.wind_encoder.out_features diff --git a/pvnet/models/multimodal/site_encoders/encoders.py b/pvnet/models/multimodal/site_encoders/encoders.py index b82c9537..cd5e725f 100644 --- a/pvnet/models/multimodal/site_encoders/encoders.py +++ b/pvnet/models/multimodal/site_encoders/encoders.py @@ -391,6 +391,7 @@ def forward(self, x): return x_out + class SinglePVAttentionNetwork(AbstractPVSitesEncoder): """A simple attention-based model with a single multihead attention layer