Merge pull request #169 from openclimatefix/update-plot-features-filters

Include all PVNet models in plot features
openclimatefix · Jul 9, 2024 · 052a6dc · 052a6dc
2 parents 96f0b63 + 924aac4
commit 052a6dc
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/plots/utils.py b/src/plots/utils.py
@@ -80,7 +80,7 @@ def model_is_probabilistic(model_name):
     is_prob = (
         (model_name in ["National_xg", "blend"])
         or
-        (model_name.startswith("pvnet_v2") and not model_name.endswith("_gsp_sum"))
+        (model_name.startswith("pvnet") and not model_name.endswith("_gsp_sum"))
     )
     return is_prob
 
@@ -90,7 +90,7 @@ def model_is_gsp_regional(model_name):
     is_regional = (
         (model_name=="cnn")
         or
-        (model_name.startswith("pvnet_v2") and not model_name.endswith("_gsp_sum"))
+        (model_name.startswith("pvnet") and not model_name.endswith("_gsp_sum"))
     )
     return is_regional