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