convert sigs to numpy array

datasig-ac-uk · Oct 24, 2023 · 571c34a · 571c34a
1 parent 06a4a8a
commit 571c34a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/signature_mahalanobis_knn/sig_mahal_knn.py b/src/signature_mahalanobis_knn/sig_mahal_knn.py
@@ -127,7 +127,7 @@ def fit(
                 delayed(self.signature_transform.fit_transform)(X[i])
                 for i in range(len(X))
             )
-            self.signatures = pd.concat(sigs)
+            self.signatures = np.array(pd.concat(sigs))
         else:
             self.signatures = signatures
 
@@ -209,7 +209,7 @@ def conformance(
                 delayed(self.signature_transform.fit_transform)(X[i])
                 for i in range(len(X))
             )
-            signatures = pd.concat(sigs)
+            signatures = np.array(pd.concat(sigs))
 
         if isinstance(self.knn, NearestNeighbors):
             # compute KNN distances for the signatures of the data points