Add solar position for PV-sites

Author: jacobbieker

pvnet/models/multimodal/multimodal.py

@@ -342,9 +342,16 @@ def forward(self, x):
                 modes["sensor"] = self.sensor_encoder(x_tmp)
 
         if self.include_sun:
-            sun = torch.cat(
-                (x[BatchKey.gsp_solar_azimuth], x[BatchKey.gsp_solar_elevation]), dim=1
-            ).float()
+            if self.target_key_name == "gsp":
+                sun = torch.cat(
+                    (x[BatchKey.gsp_solar_azimuth], x[BatchKey.gsp_solar_elevation]), dim=1
+                ).float()
+            elif self.target_key_name == "pv":
+                sun = torch.cat(
+                    (x[BatchKey.pv_solar_azimuth], x[BatchKey.pv_solar_elevation]), dim=1
+                ).float()
+            else:
+                raise ValueError(f"Unknown target key for sun elevation {self.target_key_name}")
             sun = self.sun_fc1(sun)
             modes["sun"] = sun