From 4a6467c9b3f4b62178e060a2a22738d3a7a17750 Mon Sep 17 00:00:00 2001
From: JalenCato <jalencato23pistons@gmail.com>
Date: Wed, 11 Oct 2023 23:14:35 +0000
Subject: [PATCH] add e2e test

---
 tests/end2end-tests/data_process/data_gen.py | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/tests/end2end-tests/data_process/data_gen.py b/tests/end2end-tests/data_process/data_gen.py
index abab89ea3d..7173048635 100644
--- a/tests/end2end-tests/data_process/data_gen.py
+++ b/tests/end2end-tests/data_process/data_gen.py
@@ -221,6 +221,13 @@ def split_data(data, num):
                 "feature_name": "feat_fp16",
                 "out_dtype": 'float16',
             },
+            {
+                "feature_col": "float2",
+                "feature_name": "feat_bucket",
+                "out_dtype": 'float16',
+                "transform": {"name": "bucket",
+                              "bucket": [100, 200, 300, 1000]}
+            }
         ],
         "labels":       [
             {