remove reshape from (n,d,m) to (n,m,d) in numba utils

sktime · Apr 9, 2022 · f5e81ff · f5e81ff
1 parent 2120451
commit f5e81ff
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/sktime/distances/_numba_utils.py b/sktime/distances/_numba_utils.py
@@ -136,6 +136,9 @@ def to_numba_pairwise_timeseries(x: np.ndarray) -> np.ndarray:
     if num_dims == 1:
         shape = _x.shape
         _x = np.reshape(_x, (1, 1, shape[0]))
+    elif num_dims == 2:
+        shape = _x.shape
+        _x = np.reshape(_x, (1, shape[0], shape[1]))
     elif num_dims > 3:
         raise ValueError(
             "The matrix provided has more than 3 dimensions. This is not"