Fix tensorarray to numpy conversion (ray-project#34115)

* Revert "[Datasets] Revert "Enable streaming executor by default (ray-project#32493)" (ray-project#33485)" This reverts commit 5c79954. * Fix tensorarray to numpy conversion Signed-off-by: Jack He <[email protected]>
ProjectsByJackHe · May 4, 2023 · bcc6353 · bcc6353
1 parent 8948628
commit bcc6353
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/ray/serve/air_integrations.py b/python/ray/serve/air_integrations.py
@@ -74,7 +74,7 @@ def _unpack_dataframe_to_serializable(output_df: "pd.DataFrame") -> "pd.DataFram
     for col in output_df.columns:
         # TensorArray requires special handling to numpy array.
         if isinstance(output_df.dtypes[col], TensorDtype):
-            output_df.loc[:, col] = list(output_df[col].to_numpy())
+            output_df[col] = list(output_df[col].to_numpy())
         # # DL predictor outputs raw ndarray outputs as opaque numpy object.
         # # ex: output_df = pd.DataFrame({"predictions": [np.array(1)]})
         elif output_df.dtypes[col] == np.dtype(object) and all(