if PLive in name, dont split

openclimatefix · Feb 16, 2024 · 827ce61 · 827ce61
1 parent 5f5fd56
commit 827ce61
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/plots/utils.py b/src/plots/utils.py
@@ -34,6 +34,8 @@ def get_colour_from_model_name(model_name, opacity=1.0):
     # get colour from model, some models have a space and a datetime
     model_name_only = model_name.split(" ")[0]
     colour = colour_per_model.get(model_name_only, "#FFFFFF")
+    if "PVLive" in model_name:
+        colour = colour_per_model.get(model_name, "#FFFFFF")
 
     # change opacity to hex
     rgb = hex_to_rgb(colour)