Skip to content

Commit 3cb800f

Browse files
committed
fix: test
1 parent f396d68 commit 3cb800f

File tree

1 file changed

+2
-6
lines changed

1 file changed

+2
-6
lines changed

src/codeflare_sdk/ray/rayjobs/rayjob.py

Lines changed: 2 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -194,14 +194,10 @@ def _build_rayjob_cr(self) -> Dict[str, Any]:
194194
rayjob_cr["spec"]["activeDeadlineSeconds"] = self.active_deadline_seconds
195195

196196
# Add entrypoint resource requirements if specified
197-
entrypoint_resources = {}
198197
if self.entrypoint_num_cpus is not None:
199-
entrypoint_resources["cpu"] = str(self.entrypoint_num_cpus)
198+
rayjob_cr["spec"]["entrypointNumCpus"] = self.entrypoint_num_cpus
200199
if self.entrypoint_num_gpus is not None:
201-
entrypoint_resources["gpu"] = str(self.entrypoint_num_gpus)
202-
203-
if entrypoint_resources:
204-
rayjob_cr["spec"]["entrypointResources"] = entrypoint_resources
200+
rayjob_cr["spec"]["entrypointNumGpus"] = self.entrypoint_num_gpus
205201

206202
# Add runtime environment if specified
207203
if self.runtime_env:

0 commit comments

Comments
 (0)