diff --git a/.buildkite/rllib.rayci.yml b/.buildkite/rllib.rayci.yml
index 3072ffc3998f..860ec5ae3f63 100644
--- a/.buildkite/rllib.rayci.yml
+++ b/.buildkite/rllib.rayci.yml
@@ -96,7 +96,9 @@ steps:
 
   - label: ":brain: rllib: data tests"
     if: build.branch != "master"
-    tags: data
+    tags:
+      - data
+      - rllib
     instance_type: large
     commands:
       # learning tests pytorch
@@ -106,14 +108,6 @@ steps:
         --except-tags multi_gpu,gpu,tf_only,tf2_only
         --test-arg --framework=torch
 
-      # learning tests tF2
-      - bazel run //ci/ray_ci:test_in_docker -- //rllib/... rllib 
-        --parallelism-per-worker 3
-        --only-tags learning_tests_with_ray_data
-        --except-tags multi_gpu,gpu,torch_only
-        --test-arg --framework=tf2
-        --skip-ray-installation # reuse the same docker image as the previous run
-
       # rllib unittests
       - bazel run //ci/ray_ci:test_in_docker -- //rllib/... rllib 
         --parallelism-per-worker 3