fix pipeline issues (#41487)

- fix data test dependencies
- fix arm64 image building
- add missed version change

Signed-off-by: Lonnie Liu <lonnie@anyscale.com>

---------

Signed-off-by: Lonnie Liu <lonnie@anyscale.com>

Author: aslonnie

.buildkite/data.rayci.yml

@@ -13,7 +13,7 @@ steps:
         --build-name data6build
         --test-env RAY_DATA_USE_STREAMING_EXECUTOR=1
         --except-tags data_integration,doctest
-    depends_on: data6build
+    depends_on: [ "data6build", "forge" ]
     job_env: forge
 
   - label: ":database: data: arrow 12 tests"
@@ -29,7 +29,7 @@ steps:
         --build-name data12build
         --test-env RAY_DATA_USE_STREAMING_EXECUTOR=1
         --except-tags data_integration,doctest
-    depends_on: data12build
+    depends_on: [ "data12build", "forge" ]
     job_env: forge
 
   - label: ":database: data: arrow nightly tests"
@@ -45,7 +45,7 @@ steps:
         --build-name datanbuild
         --test-env RAY_DATA_USE_STREAMING_EXECUTOR=1
         --except-tags data_integration,doctest
-    depends_on: datanbuild
+    depends_on: [ "datanbuild", "forge" ]
     job_env: forge
 
   - label: ":database: data: flaky tests"
@@ -58,5 +58,5 @@ steps:
       - 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
+    depends_on: [ "data12build", "forge" ]
     job_env: forge

ci/docker/data.build.Dockerfile

@@ -14,4 +14,3 @@ SHELL ["/bin/bash", "-ice"]
 COPY . .
 
 RUN DATA_PROCESSING_TESTING=1 ARROW_VERSION=$ARROW_VERSION ./ci/env/install-dependencies.sh
-RUN pip install "datasets==2.14.0"

java/pom.xml

@@ -5,7 +5,7 @@
   <modelVersion>4.0.0</modelVersion>
   <groupId>io.ray</groupId>
   <artifactId>ray-superpom</artifactId>
-  <version>2.8.0</version>
+  <version>2.8.1</version>
   <packaging>pom</packaging>
   <name>Ray Project Parent POM</name>
   <description>An open source framework that provides a simple, universal API for building distributed applications.

python/requirements/test-requirements.txt

@@ -75,7 +75,6 @@ werkzeug==2.1.2
 xlrd==2.0.1
 yq==3.2.2
 memray; platform_system != "Windows" and sys_platform != "darwin" and platform_machine != 'aarch64'
-memray @ git+https://github.com/bloomberg/memray.git; platform_system != "Windows" and sys_platform != "darwin" and platform_machine == 'aarch64'
 
 # For doc tests
 myst-parser==0.15.2