Merge pull request #146 from openclimatefix/issue/legacy-gsp-id

add loggin, and unsqueeze for new models
openclimatefix · Oct 18, 2024 · 63a5e89 · 63a5e89
2 parents 06b66bf + 04eb0c9
commit 63a5e89
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/pvnet_app/forecast_compiler.py b/pvnet_app/forecast_compiler.py
@@ -140,6 +140,7 @@ def predict_batch(self, batch: NumpyBatch) -> None:
         these_gsp_ids = batch[BatchKey.gsp_id].cpu().numpy()
         self.gsp_ids_each_batch += [these_gsp_ids]
 
+        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)
 

diff --git a/pvnet_app/model_configs/all_models.yaml b/pvnet_app/model_configs/all_models.yaml
@@ -51,10 +51,10 @@ models:
   - name: pvnet_v2
     pvnet:
       repo: openclimatefix/pvnet_uk_region
-      version: 62e5e20ab793cee7cf94eadac870d2199501a730
+      version: aa73cdafd1db8df3c8b7f5ecfdb160989e7639ac
     summation:
       repo: openclimatefix/pvnet_v2_summation
-      version: ffac655f9650b81865d96023baa15839f3ce26ec
+      version: a7fd71727f4cb2b933992b2108638985e24fa5a3
     use_adjuster: True
     save_gsp_sum: False
     verbose: True