diff --git a/mta_manager/mta.py b/mta_manager/mta.py index 3b114b6..d91707d 100644 --- a/mta_manager/mta.py +++ b/mta_manager/mta.py @@ -5,14 +5,14 @@ import json from google.transit import gtfs_realtime_pb2 from protobuf_to_dict import protobuf_to_dict from time import time -from train import Train +from .train import Train class MTA(object): # Create a data filter object. # Then be able to update that object on the fly. # 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, endpoints_file="./endpoints.json", callback_frequency=10, max_arrival_time=30): self.header = { diff --git a/mta_manager/train.py b/mta_manager/train.py index c4d66a4..f8dc756 100644 --- a/mta_manager/train.py +++ b/mta_manager/train.py @@ -1,4 +1,4 @@ -from stop import Stop +from .stop import Stop class Train(object): def __init__(self, id, route, stops):