fix gear if gear is NaN (#425)

b4mad · Sep 5, 2023 · af61153 · af61153
1 parent a4432b2
commit af61153
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/components/paddock/telemetry/fast_lap_analyzer.py b/components/paddock/telemetry/fast_lap_analyzer.py
@@ -1,6 +1,8 @@
 import logging
 import statistics
 
+import numpy as np
+
 from .analyzer import Analyzer
 from .influx import Influx
 from .models import FastLap
@@ -224,7 +226,7 @@ def gear_features(self, df):
         # create a dict which maps the distance to the gear
         distance_gear = dict(zip(distances, gears))
         return {
-            "gear": int(gear),
+            "gear": int(gear) if not np.isnan(gear) else 0,
             "distance_gear": distance_gear,
         }