Skip to content

Commit

Permalink
Update
Browse files Browse the repository at this point in the history
  • Loading branch information
felix-e-h-p committed Mar 11, 2025
1 parent b842dad commit 26ce76c
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions pvnet/models/multimodal/multimodal_base.py
Original file line number Diff line number Diff line change
Expand Up @@ -43,16 +43,16 @@ def _adapt_batch(self, batch):
if self.include_sun:
sun_len = self.forecast_len + self.history_len + 1

# Check for new standalone solar position keys first
solar_position_keys = []
# Slife off end of solar coords
for s in ["solar_azimuth", "solar_elevation"]:
if s in batch.keys():
solar_position_keys.append(s)
batch[s] = batch[s][:, :sun_len]

# Check for legacy keys if new keys aren't found
if not solar_position_keys:
# Slice off the end of the legacy solar coords data
# Slice off the end of the legacy solar coords
for s in ["solar_azimuth", "solar_elevation"]:
key = f"{self._target_key}_{s}"
if key in batch.keys():
Expand Down

0 comments on commit 26ce76c

Please sign in to comment.