Skip to content
Snippets Groups Projects

Draft: Develop

Open Raphael Sturgis requested to merge develop into main
@@ -358,11 +358,11 @@ class AISTrajectory(AISPoints):
elif type(features) is list:
features_vectors = self.df[features].to_numpy()
for c in features_vectors.T:
bounds.append((0, max(c)))
bounds.append([0, max(c)])
elif type(features) is str:
features_vectors = self.df[[features]].to_numpy()
for c in features_vectors.T:
bounds.append((0, max(c)))
bounds.append([0, max(c)])
elif type(features) is dict:
bounds = list(features.values())
features_vectors = self.df[features.keys()].to_numpy()
@@ -383,6 +383,6 @@ class AISTrajectory(AISPoints):
generate_links(data, positions, height, width, lower_lat, upper_lat, lower_lon, upper_lon)
else:
generate_points_with_features(data, positions, features_vectors, bounds, node_size, height, width,
generate_points_with_features(data, positions, features_vectors, np.array(bounds), node_size, height, width,
lower_lat, upper_lat, lower_lon, upper_lon, link)
return data
Loading