diff --git a/DataImporter.py b/DataImporter.py index 5d4d413985268fa05a806bde3c9d104808e162df..0c49463dac2b96cdce867de42a02ab73b00ec41e 100644 --- a/DataImporter.py +++ b/DataImporter.py @@ -26,15 +26,14 @@ class DataImporter(DataHandler, ABC): x = 0 - def importRaceSessions(self, races: list[Session], laps = True, telemetry = False, weather = True, messages = False): + def importRaceSessions(self, sessionIdentifiers: list[(int, int)], laps = True, telemetry = False, weather = True, messages = False): sessions: list[Session] = [] - for raceSession in races: - sessions.append(self.importSession(raceSession, "R", laps, weather, messages, telemetry)) + for sessionIdentifier in sessionIdentifiers: + sessions.append(self.importSession(sessionIdentifier, "R", laps, weather, messages, telemetry)) return sessions - def importSession(self, session: Session, sessionType: str, laps = True, telemetry = False, weather = True, messages = False): - season, raceIndex = session - session = fastf1.get_session(season, raceIndex, sessionType) + def importSession(self, sessionIdentifier: (int, int), sessionType: str, laps = True, telemetry = False, weather = True, messages = False): + session = fastf1.get_session(sessionIdentifier[0], sessionIdentifier[1], sessionType) session.load(laps = laps, telemetry = telemetry, weather = weather, messages = messages) return session \ No newline at end of file