Merge pull request #242 from openclimatefix/fix-time

fix issue with time features missing in forward multimodal.py
openclimatefix · Aug 7, 2024 · d87acd0 · d87acd0
2 parents 9a7e70e + 9677877
commit d87acd0
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/pvnet/models/multimodal/multimodal.py b/pvnet/models/multimodal/multimodal.py
@@ -401,6 +401,19 @@ def forward(self, x):
             sun = self.sun_fc1(sun)
             modes["sun"] = sun
 
+        if self.include_time:
+            time = torch.cat(
+                (
+                    x[BatchKey[f"{self._target_key_name}_date_sin"]],
+                    x[BatchKey[f"{self._target_key_name}_date_cos"]],
+                    x[BatchKey[f"{self._target_key_name}_time_sin"]],
+                    x[BatchKey[f"{self._target_key_name}_time_cos"]],
+                ),
+                dim=1,
+            ).float()
+            time = self.time_fc1(time)
+            modes["time"] = time
+
         out = self.output_network(modes)
 
         if self.use_quantile_regression: