use lowercase key for plotting

starsimhub · Nov 18, 2024 · 0ad7941 · 0ad7941
1 parent d77cede
commit 0ad7941
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/starsim/sim.py b/starsim/sim.py
@@ -598,9 +598,9 @@ def plot(self, key=None, fig=None, style='fancy', show_data=True, show_skipped=F
 
             if key is not None:
                 if isinstance(key, str):
-                    flat = {k:v for k,v in flat.items() if (key in k)}
+                    flat = {k:v for k,v in flat.items() if (key.lower() in k)}
                 else:
-                    flat = {k:flat[k] for k in key}
+                    flat = {k.lower():flat[k.lower()] for k in key}
 
             # Get the figure
             if fig is None: