File tree Expand file tree Collapse file tree 1 file changed +2
-6
lines changed
src/codeflare_sdk/ray/rayjobs Expand file tree Collapse file tree 1 file changed +2
-6
lines changed Original file line number Diff line number Diff 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 :
You can’t perform that action at this time.
0 commit comments