refactor driver team mapping

theOehrly · Mar 29, 2024 · 36d7b03 · 36d7b03
1 parent 292cab3
commit 36d7b03
Show file tree

Hide file tree

Showing 2 changed files with 211 additions and 170 deletions.
diff --git a/fastf1/plotting/_constants/base.py b/fastf1/plotting/_constants/base.py
@@ -1,4 +1,7 @@
-from enum import Enum
+from enum import (
+    Enum,
+    EnumType
+)
 from typing import Dict
 
 
@@ -21,5 +24,5 @@ class Compounds(Enum):
 class BaseSeason:
     Colormaps: Dict[Colormaps, dict]
     CompoundColors: Dict[Compounds, str]
-    ShortTeamNames: Dict[str, str]
-    Teams: Enum
+    ShortTeamNames: Dict[EnumType, str]
+    Teams: EnumType