[ENH] Add COL to driver dictionaries (#609)

(cherry picked from commit 1dc86a0)
theOehrly · Jul 5, 2024 · e96f931 · e96f931
1 parent 8040300
commit e96f931
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/fastf1/events.py b/fastf1/events.py
@@ -882,7 +882,7 @@ def __init__(self, *args, year: int = 0,
             if self[col].isna().all():
                 if _type == 'datetime64[ns]':
                     self[col] = pd.NaT
-                elif _type == object:
+                elif _type == object:  # noqa: E721, type comparison with ==
                     self[col] = None
                 else:
                     self[col] = _type()

diff --git a/fastf1/plotting.py b/fastf1/plotting.py
@@ -144,6 +144,7 @@ def warn_change(self):
     "alexander albon": "#005aff",
     "logan sargeant": "#012564",
     "zak osullivan": "#1b3d97",
+    "franco colapinto": "#639aff"
 }
 """Mapping of driver names to driver colors (hex color codes).
 (current season only)"""
@@ -169,7 +170,7 @@ def warn_change(self):
     'HAM': 'lewis hamilton', 'RUS': 'george russell',
     'VES': 'frederik vesti',
     'ALB': 'alexander albon', 'SAR': 'logan sargeant',
-    'OSU': 'zak osullivan'}
+    'OSU': 'zak osullivan', 'COL': 'franco colapinto'}
 """Mapping of driver names to theirs respective abbreviations."""
 
 COMPOUND_COLORS: Dict[str, str] = {