Update main.py

main.py

@@ -153,6 +153,8 @@ def telemetry_data(year: int, event: str | int, session: str, driver: str, lap_n
     
     laptime = selected_lap.LapTime.values[0]
     data_key = f"{driver} - Lap {int(lap_number)} - {year} {session} [{int(laptime//60)}:{laptime%60}]"
+
+    telemetry['DRS'] = telemetry['DRS'].apply(lambda x: 1 if x in [10,12,14] else 0)
     
     brake_tel = []
     drs_tel = []
@@ -171,7 +173,7 @@ def telemetry_data(year: int, event: str | int, session: str, driver: str, lap_n
         brake_tel.append(brake)        
         
         drs = {"x": row['Distance'],
-                    "y": row['DRS'],
+                    "y": 1 if row['DRS'],
                     }
         drs_tel.append(drs)