Skip to content

Commit 6f56f55

Browse files
committed
RHOAIENG-39075: Add new E2E test structure and initial existing cluster test
1 parent b713531 commit 6f56f55

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

53 files changed

+5272
-6
lines changed

src/codeflare_sdk/ray/cluster/build_ray_cluster.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -308,7 +308,7 @@ def get_head_container_spec(
308308
head_container = V1Container(
309309
name="ray-head",
310310
image=update_image(cluster.config.image),
311-
image_pull_policy="Always",
311+
image_pull_policy="IfNotPresent",
312312
ports=[
313313
V1ContainerPort(name="gcs", container_port=6379),
314314
V1ContainerPort(name="dashboard", container_port=8265),
@@ -357,7 +357,7 @@ def get_worker_container_spec(
357357
worker_container = V1Container(
358358
name="machine-learning",
359359
image=update_image(cluster.config.image),
360-
image_pull_policy="Always",
360+
image_pull_policy="IfNotPresent",
361361
lifecycle=V1Lifecycle(
362362
pre_stop=V1LifecycleHandler(
363363
_exec=V1ExecAction(["/bin/sh", "-c", "ray stop"])

src/codeflare_sdk/ray/cluster/config.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -98,9 +98,9 @@ class ClusterConfiguration:
9898

9999
name: str
100100
namespace: Optional[str] = None
101-
head_cpu_requests: Union[int, str] = 2
101+
head_cpu_requests: Union[int, str] = 1
102102
head_cpu_limits: Union[int, str] = 2
103-
head_memory_requests: Union[int, str] = 8
103+
head_memory_requests: Union[int, str] = 5
104104
head_memory_limits: Union[int, str] = 8
105105
head_extended_resource_requests: Dict[str, Union[str, int]] = field(
106106
default_factory=dict
@@ -109,8 +109,8 @@ class ClusterConfiguration:
109109
worker_cpu_requests: Union[int, str] = 1
110110
worker_cpu_limits: Union[int, str] = 1
111111
num_workers: int = 1
112-
worker_memory_requests: Union[int, str] = 2
113-
worker_memory_limits: Union[int, str] = 2
112+
worker_memory_requests: Union[int, str] = 3
113+
worker_memory_limits: Union[int, str] = 6
114114
worker_tolerations: Optional[List[V1Toleration]] = None
115115
appwrapper: bool = False
116116
envs: Dict[str, str] = field(default_factory=dict)

tests/e2e_v2/__init__.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,2 @@
1+
# E2E Test Suite v2
2+
# Restructured pytest-based E2E tests for CodeFlare SDK

tests/e2e_v2/cluster_management/__init__.py

Whitespace-only changes.

tests/e2e_v2/cluster_management/configuration/__init__.py

Whitespace-only changes.

tests/e2e_v2/cluster_management/configuration/test_advanced.py

Whitespace-only changes.

tests/e2e_v2/cluster_management/configuration/test_heterogeneous.py

Whitespace-only changes.

tests/e2e_v2/cluster_management/configuration/test_images.py

Whitespace-only changes.

tests/e2e_v2/cluster_management/configuration/test_resources.py

Whitespace-only changes.

tests/e2e_v2/cluster_management/configuration/test_volumes.py

Whitespace-only changes.

0 commit comments

Comments
 (0)