Error mapping teamcolors to driver

[walteij]

I'm trying to plot racepace for all drivers in a single plot. However, I'm getting an error when trying to map the drivers to their teams for coloring purposes, the script I'm using as a base can be found here at medium.com:

for driver in drivers_to_visualize:
    driver_laps = laps.pick_driver(driver)[['LapNumber', 'LapTimeSeconds', 'Team']]
       
    # Select all the laps from that driver
    driver_laps = driver_laps.dropna()
    
    # Extract the team for coloring purposes
    team = pd.unique(driver_laps['Team'])[0]
        
    # X-coordinate is the lap number
    x = driver_laps['LapNumber']

The error thrown is:

File "C:\path\to\plots\telemetry.py", line 207, in racepace
    team = pd.unique(driver_laps['Team'])[0]
IndexError: index 0 is out of bounds for axis 0 with size

The annoying part is that when I run the script separately, there's no issue. But when I try and add my code as a module for my discord bot, the error above is thrown. An tips or clues to get me looking in the right direction?

[walteij]

Hmm, found another thing that seems to be related. I tried to use the qualification result from the documentation, and here I'm getting a different error, but it is related to the team color mapping. I am using version = '2.2.4'

Traceback (most recent call last):
  File "C:\Users\WaltervanEijk\OneDrive - BORG-OSC\Documenten\DiscordBot\plots\old\plot_qualifying_results (4).py", line 79, in <module>
    color = fastf1.plotting.team_color(lap['Team'])
  File "C:\Users\WaltervanEijk\AppData\Roaming\Python\Python39\site-packages\fastf1\plotting.py", line 182, in team_color
    if identifier.upper() in TEAM_TRANSLATE:
AttributeError: 'NaTType' object has no attribute 'upper'

[walteij]

Found the answer, simple programming error from my side.
I'm creating the drivers_to_visualise with all drivers on track, where my laps are only for drivers that used SOFT tyres. So I'm getting a null result from that.
By creating the drivers_to_visualise from that only have the drivers using a specific compound, I was able to fix this.