jail //python/ray/data:test_streaming_executor

Signed-off-by: can <can@anyscale.com>
ray-project · Sep 8, 2023 · cd98987 · cd98987
1 parent 6523d94
commit cd98987
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 1 deletion.
diff --git a/.buildkite/data.rayci.yml b/.buildkite/data.rayci.yml
@@ -51,3 +51,13 @@ steps:
         --except-tags data_integration,doctest
     depends_on: databbuild
     job_env: databbuild
+
+  - label: ":database: data: flaky tests"
+    instance_type: medium
+    soft_fail: true
+    commands:
+      - bazel run //ci/ray_ci:test_in_docker -- //... data --run-flaky-tests 
+        --test-env RAY_DATA_USE_STREAMING_EXECUTOR=1 --parallelism-per-worker 3
+        --build-name data12build
+    depends_on: data12build
+    job_env: data12build
diff --git a/ci/ray_ci/data.tests.yml b/ci/ray_ci/data.tests.yml
@@ -1 +1,5 @@
-flaky_tests: []
+flaky_tests:
+  - //python/ray/data:test_streaming_executor
+  - //python/ray/data:test_split
+  - //python/ray/data:test_object_gc
+  - //python/ray/data:test_stats