change plot_plan to take lists and numpy arrays.

This commit is contained in:
Cory Balaton 2024-10-01 19:16:33 +02:00
parent b1c6ad01a0
commit 2ee3f59702
Signed by: coryab
GPG Key ID: F7562F0EC4E4A61B

View File

@ -1,4 +1,4 @@
from typing import Dict, List, Tuple from typing import Dict, List, Tuple, Union
import matplotlib.pyplot as plt import matplotlib.pyplot as plt
import numpy as np import numpy as np
@ -33,7 +33,7 @@ city_coords: Dict = {
} }
def plot_plan(city_order: List[str]) -> None: def plot_plan(city_order: Union[List[str], npt.NDArray]) -> None:
"""A function that plots the circuit given by city_order. """A function that plots the circuit given by city_order.
This function was given in the assignment from 2024. This function was given in the assignment from 2024.