Alter is prob check

openclimatefix · Jun 13, 2024 · 0e2bc54 · 0e2bc54
1 parent 42cf739
commit 0e2bc54
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/forecast.py b/src/forecast.py
@@ -65,7 +65,7 @@ def forecast_page():
             is_prob = (
                 (model in ["National_xg", "blend"])
                 or
-                (model.startswith("pvnet_v2") and not model.endswith("_gsp_sum"))
+                (model.startswith("pvnet") and not model.endswith("_gsp_sum"))
             )
             if is_prob:
                 selected_prob_models.append(model)