fixing imports.

This commit is contained in:
Lucas Oskorep
2022-09-13 17:12:40 -04:00
parent 30cbe70d7b
commit 498657cbdd
2 changed files with 3 additions and 3 deletions

View File

@@ -5,14 +5,14 @@ import json
from google.transit import gtfs_realtime_pb2 from google.transit import gtfs_realtime_pb2
from protobuf_to_dict import protobuf_to_dict from protobuf_to_dict import protobuf_to_dict
from time import time from time import time
from train import Train from .train import Train
class MTA(object): class MTA(object):
# Create a data filter object. # Create a data filter object.
# Then be able to update that object on the fly. # Then be able to update that object on the fly.
# This filter should return all possible trains and stations by default. # This filter should return all possible trains and stations by default.
# If anyhting is added it gets filtered out. # If anything is added it gets filtered out.
def __init__(self, api_key: str, routes, station_ids, timing_callbacks=None, alert_callbacks=None, def __init__(self, api_key: str, routes, station_ids, timing_callbacks=None, alert_callbacks=None,
endpoints_file="./endpoints.json", callback_frequency=10, max_arrival_time=30): endpoints_file="./endpoints.json", callback_frequency=10, max_arrival_time=30):
self.header = { self.header = {

View File

@@ -1,4 +1,4 @@
from stop import Stop from .stop import Stop
class Train(object): class Train(object):
def __init__(self, id, route, stops): def __init__(self, id, route, stops):