fix e2e test

awslabs · Oct 18, 2023 · bb1b118 · bb1b118
1 parent 51989fc
commit bb1b118
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/tests/end2end-tests/data_process/data_gen.py b/tests/end2end-tests/data_process/data_gen.py
@@ -226,7 +226,10 @@ def split_data(data, num):
                 "feature_name": "feat_bucket",
                 "out_dtype": 'float16',
                 "transform": {"name": "bucket",
-                              "bucket": [100, 200, 300, 1000]}
+                              "range": [10, 50],
+                              "bucket_cnt": 2,
+                              "slide_window_size": 10
+}
             }
         ],
         "labels":       [