chore: fix typing of array in clusterer

crux-bphc · Jul 30, 2023 · 54d1144 · 54d1144
1 parent f369f54
commit 54d1144
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/clusterer/datapoint_clusterer.py b/clusterer/datapoint_clusterer.py
@@ -87,7 +87,7 @@ def create_clusters(_datapoints: list[Datapoint], _epsilon:float, _min_samples:i
     Returns:
         list[Cluster] : a list of clusters found
     """
-    points = np.array([point.data for point in _datapoints]) 
+    points = np.array([point.data.numpy() for point in _datapoints]) 
     scaler = StandardScaler()
     scaler.fit(points)
     points = scaler.transform(points)