diff --git a/pyrmv/methods/board_arrival.py b/pyrmv/methods/board_arrival.py index 72db566..0d49a8c 100644 --- a/pyrmv/methods/board_arrival.py +++ b/pyrmv/methods/board_arrival.py @@ -49,6 +49,9 @@ def board_arrival( * BoardArrival: Instance of BoardArrival object. """ + if (isinstance(direction, Stop) or isinstance(direction, StopTrip)): + direction = direction.id + board_raw = raw_board_arrival( accessId=access_id, id=id, diff --git a/pyrmv/methods/board_departure.py b/pyrmv/methods/board_departure.py index a8e4bf7..da7bb90 100644 --- a/pyrmv/methods/board_departure.py +++ b/pyrmv/methods/board_departure.py @@ -49,6 +49,9 @@ def board_departure( * BoardDeparture: Instance of BoardDeparture object. """ + if (isinstance(direction, Stop) or isinstance(direction, StopTrip)): + direction = direction.id + board_raw = raw_board_departure( accessId=access_id, id=id,