This is an automated email from the ASF dual-hosted git repository.
shahar pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/airflow.git
The following commit(s) were added to refs/heads/main by this push:
new 02c1cae838c Fix ray job system test (#61848)
02c1cae838c is described below
commit 02c1cae838cd34dfc4a030d8a22f528e3e908791
Author: VladaZakharova <[email protected]>
AuthorDate: Sun Feb 15 06:52:09 2026 +0100
Fix ray job system test (#61848)
---
.../system/google/cloud/ray/example_ray_job.py | 3 +-
.../system/google/cloud/ray/resources/__init__.py | 16 -------
.../system/google/cloud/ray/resources/heavy.py | 54 ----------------------
3 files changed, 1 insertion(+), 72 deletions(-)
diff --git a/providers/google/tests/system/google/cloud/ray/example_ray_job.py
b/providers/google/tests/system/google/cloud/ray/example_ray_job.py
index be3aa637b95..f4a5533fb87 100644
--- a/providers/google/tests/system/google/cloud/ray/example_ray_job.py
+++ b/providers/google/tests/system/google/cloud/ray/example_ray_job.py
@@ -90,9 +90,8 @@ with DAG(
submit_ray_job = RaySubmitJobOperator(
task_id="submit_ray_job",
cluster_address="{{
task_instance.xcom_pull(task_ids='get_ray_cluster')['dashboard_address'] }}",
- entrypoint="python3 heavy.py",
+ entrypoint="echo hi && sleep 105 && echo hi2",
runtime_env={
- "working_dir":
"./providers/google/tests/system/google/cloud/ray/resources",
"pip": [
"ray==2.33.0",
],
diff --git
a/providers/google/tests/system/google/cloud/ray/resources/__init__.py
b/providers/google/tests/system/google/cloud/ray/resources/__init__.py
deleted file mode 100644
index 13a83393a91..00000000000
--- a/providers/google/tests/system/google/cloud/ray/resources/__init__.py
+++ /dev/null
@@ -1,16 +0,0 @@
-# Licensed to the Apache Software Foundation (ASF) under one
-# or more contributor license agreements. See the NOTICE file
-# distributed with this work for additional information
-# regarding copyright ownership. The ASF licenses this file
-# to you under the Apache License, Version 2.0 (the
-# "License"); you may not use this file except in compliance
-# with the License. You may obtain a copy of the License at
-#
-# http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing,
-# software distributed under the License is distributed on an
-# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
-# KIND, either express or implied. See the License for the
-# specific language governing permissions and limitations
-# under the License.
diff --git a/providers/google/tests/system/google/cloud/ray/resources/heavy.py
b/providers/google/tests/system/google/cloud/ray/resources/heavy.py
deleted file mode 100644
index fb25b725a58..00000000000
--- a/providers/google/tests/system/google/cloud/ray/resources/heavy.py
+++ /dev/null
@@ -1,54 +0,0 @@
-# Licensed to the Apache Software Foundation (ASF) under one
-# or more contributor license agreements. See the NOTICE file
-# distributed with this work for additional information
-# regarding copyright ownership. The ASF licenses this file
-# to you under the Apache License, Version 2.0 (the
-# "License"); you may not use this file except in compliance
-# with the License. You may obtain a copy of the License at
-#
-# http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing,
-# software distributed under the License is distributed on an
-# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
-# KIND, either express or implied. See the License for the
-# specific language governing permissions and limitations
-# under the License.
-from __future__ import annotations
-
-import time
-
-import ray
-
-# Initialize Ray
-ray.init()
-
-
-# Define a computationally intensive task
[email protected](num_cpus=1)
-def heavy_task(x):
- """
- Simulates a heavy workload by performing a CPU-bound operation.
- This example calculates the sum of squares for a range of numbers.
- """
- total = 0
- for i in range(x):
- total += i * i
- time.sleep(1) # Simulate some work duration
- return total
-
-
-# Generate a large number of tasks
-num_tasks = 1000
-results = []
-for _i in range(num_tasks):
- results.append(heavy_task.remote(1000000))
-
-# Retrieve results (this will trigger autoscaling if needed)
-outputs = ray.get(results)
-
-# Print the sum of the results (optional)
-print(f"Sum of results: {sum(outputs)}")
-
-# Terminate the process
-ray.shutdown()
