diff --git a/engine.py b/engine.py index 50777ae..cf5da69 100755 --- a/engine.py +++ b/engine.py @@ -113,7 +113,7 @@ def get_dist_series_list(brand: str, n_locs: int): # split the distances matrix into a list of series, which allows us to sort each row dist_series_list = [] for i in dist_matrix: - dist_series_list.append(pd.Series(i).sort_values().head(n_locs+1)[1:]) + dist_series_list.append(pd.Series(i).sort_values().head(n_locs+1)[1:]) #ignore the 0th element (dist to itself) cached_series[brand] = (n_locs, dist_series_list)