Merge remote-tracking branch 'origin/jacob/windnet' into jacob/windnet

openclimatefix · Feb 5, 2024 · 8c9531c · 8c9531c
2 parents 771c08a + e30b0d0
commit 8c9531c
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/configs.example/model/wind_multimodal.yaml b/configs.example/model/wind_multimodal.yaml
@@ -15,7 +15,6 @@ nwp_encoders_dict:
     conv3d_channels: 32
     image_size_pixels: 16
 
-
 #--------------------------------------------
 # Sensor encoder settings
 #--------------------------------------------

diff --git a/pvnet/models/multimodal/multimodal.py b/pvnet/models/multimodal/multimodal.py
@@ -207,7 +207,9 @@ def __init__(
             if wind_history_minutes is None:
                 wind_history_minutes = history_minutes
 
-            self.wind_encoder = wind_encoder(sequence_length=wind_history_minutes // interval_minutes)
+            self.wind_encoder = wind_encoder(
+                sequence_length=wind_history_minutes // interval_minutes
+            )
 
             # Update num features
             fusion_input_features += self.wind_encoder.out_features
@@ -216,7 +218,9 @@ def __init__(
             if sensor_history_minutes is None:
                 sensor_history_minutes = history_minutes
 
-            self.sensor_encoder = sensor_encoder(sequence_length=sensor_history_minutes // sensor_interval_minutes)
+            self.sensor_encoder = sensor_encoder(
+                sequence_length=sensor_history_minutes // sensor_interval_minutes
+            )
 
             # Update num features
             fusion_input_features += self.sensor_encoder.out_features