fix the trainer code for flatten vector

Signed-off-by: ssrigiri1 <[email protected]>
numaproj · Feb 7, 2024 · ec2939d · ec2939d
1 parent 4368d34
commit ec2939d
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/numalogic/transforms/_stateless.py b/numalogic/transforms/_stateless.py
@@ -83,9 +83,8 @@ def transform(self, X: npt.NDArray[float], seq_length=20) -> npt.NDArray[float]:
                 result.extend(X[i : i + n, j])
 
         # Convert the result to a numpy array with shape (seq_length, 1)
-        result = np.array(result).reshape(-1, 1)
+        return np.array(result).reshape(-1, 1)
 
-        return result
 
     def inverse_transform(self, X: npt.NDArray[float]) -> npt.NDArray[float]:
         # Todo: Change this aswell. Fix the inverse flatten aswell