expand dims

openclimatefix · Nov 6, 2023 · bc236d7 · bc236d7
1 parent dff7225
commit bc236d7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pvnet_app/app.py b/pvnet_app/app.py
@@ -512,7 +512,7 @@ def app(
 
     # Multiply normalised forecasts by capacities and clip negatives
     logger.info(f"Converting to absolute MW using {gsp_capacities}")
-    da_abs = da_normed.clip(0, None) * gsp_capacities.values
+    da_abs = da_normed.clip(0, None) * gsp_capacities.values[: None, None]
     max_preds = da_abs.sel(output_label="forecast_mw").max(dim="target_datetime_utc")
     logger.info(f"Maximum predictions: {max_preds}")