Skip to content

Commit ad46ecc

Browse files
committed
RHOAIENG-12192 - Extend DSP e2e tests
1 parent 9e99b02 commit ad46ecc

File tree

5 files changed

+769
-24
lines changed

5 files changed

+769
-24
lines changed

ods_ci/tests/Resources/Files/pipeline-samples/v2/cache-disabled/ray_integration.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -6,11 +6,11 @@
66

77

88
# image and the sdk has a fixed value because the version matters
9-
@dsl.component(packages_to_install=["codeflare-sdk==0.21.1"], base_image=common_base_image)
9+
@dsl.component(packages_to_install=["codeflare-sdk==v0.24.0"], base_image=common_base_image)
1010
def ray_fn() -> int:
1111
import ray # noqa: PLC0415
1212
from codeflare_sdk import generate_cert # noqa: PLC0415
13-
from codeflare_sdk.cluster.cluster import Cluster, ClusterConfiguration # noqa: PLC0415
13+
from codeflare_sdk.ray.cluster import Cluster, ClusterConfiguration # noqa: PLC0415
1414

1515
cluster = Cluster(
1616
ClusterConfiguration(
@@ -24,7 +24,7 @@ def ray_fn() -> int:
2424
worker_cpu_limits=1,
2525
worker_memory_requests=1,
2626
worker_memory_limits=2,
27-
image="quay.io/modh/ray:2.35.0-py39-cu121",
27+
image="quay.io/modh/ray:2.35.0-py311-cu121",
2828
verify_tls=False
2929
)
3030
)

0 commit comments

Comments
 (0)