add e2e test

awslabs · Oct 11, 2023 · 4a6467c · 4a6467c
1 parent 0efd25e
commit 4a6467c
Showing 1 changed file with 7 additions and 0 deletions.
diff --git 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":       [
             {