From ab475be81171e3282fe3ae6e8c00dee0a88cecb0 Mon Sep 17 00:00:00 2001 From: prjanitor Date: Mon, 6 Apr 2026 21:55:08 +0300 Subject: [PATCH] Fix incorrect index calculation in _clip_clicks --- .../ali_display_ad_click/processing/prep_3_merge.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/deepray/datasets/ali_display_ad_click/processing/prep_3_merge.py b/deepray/datasets/ali_display_ad_click/processing/prep_3_merge.py index 06fb96d8..60beeaff 100644 --- a/deepray/datasets/ali_display_ad_click/processing/prep_3_merge.py +++ b/deepray/datasets/ali_display_ad_click/processing/prep_3_merge.py @@ -54,11 +54,11 @@ def _clip_clicks(row, behavior_log_cols, duration): ts_list = row[ts_col] begin_idx = 0 end_idx = len(ts_list) - for ts in ts_list: + for i, ts in enumerate(ts_list): if ts < min_ts: - begin_idx += 1 + begin_idx = max(begin_idx, i + 1) elif ts >= max_ts: - end_idx -= 1 + end_idx = min(end_idx, i) for col in cols: row[col] = row[col][begin_idx:end_idx] return row