Explicit reshape

SeldonIO · May 15, 2024 · 1f61f17 · 1f61f17
1 parent f5f6202
commit 1f61f17
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mlserver/codecs/pandas.py b/mlserver/codecs/pandas.py
@@ -26,7 +26,7 @@ def _to_series(input_or_output: InputOrOutput) -> pd.Series:
         return pd.Series(payload)
 
     if isinstance(payload, np.ndarray):
-        payload = payload.reshape(-1, 1).squeeze(axis=1)
+        payload = payload.reshape(input_or_output.shape[0], 1).squeeze(axis=1)
 
     dtype = to_dtype(input_or_output)
     return pd.Series(payload, dtype=dtype)