From 086fe55fa2ddf8769d4042cd6595a68207905ca3 Mon Sep 17 00:00:00 2001
From: nov11 <529079634@qq.com>
Date: Thu, 14 Mar 2024 23:43:05 +0800
Subject: [PATCH] fix pyarrow error

Signed-off-by: nov11 <529079634@qq.com>
---
 docs/tutorial/ranking/taobao/data/prep_3_merge.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/docs/tutorial/ranking/taobao/data/prep_3_merge.py b/docs/tutorial/ranking/taobao/data/prep_3_merge.py
index 0dc546b7..02cc440c 100644
--- a/docs/tutorial/ranking/taobao/data/prep_3_merge.py
+++ b/docs/tutorial/ranking/taobao/data/prep_3_merge.py
@@ -63,7 +63,8 @@ def _clip_clicks(row, behavior_log_cols, duration):
       elif ts >= max_ts:
         end_idx -= 1
     for col in cols:
-      row[col] = row[col][begin_idx: end_idx]
+      tmp = row[col][begin_idx: end_idx]
+      row[col] = tmp.tolist() if isinstance(tmp, np.ndarray) else tmp
   return row