fix

openclimatefix · Oct 18, 2024 · e2fd0cf · e2fd0cf
1 parent a6fda20
commit e2fd0cf
Showing 1 changed file with 3 additions and 6 deletions.
diff --git a/pvnet_app/forecast_compiler.py b/pvnet_app/forecast_compiler.py
@@ -140,12 +140,9 @@ def predict_batch(self, batch: NumpyBatch) -> None:
         these_gsp_ids = batch[BatchKey.gsp_id].cpu().numpy()
         self.gsp_ids_each_batch += [these_gsp_ids]
 
-        if self.use_legacy:
-            self.log_info(f"{batch[BatchKey.gsp_id]=}")
-        else:
-            # TODO: This change should be moved inside PVNet
-            batch[BatchKey.gsp_id] = batch[BatchKey.gsp_id].unsqueeze(1)
-
+        self.log_info(f"{batch[BatchKey.gsp_id]=}")
+        # TODO: This change should be moved inside PVNet
+        batch[BatchKey.gsp_id] = batch[BatchKey.gsp_id].unsqueeze(1)
 
         # Run batch through model
         preds = self.model(batch).detach().cpu().numpy()