@@ -160,9 +160,9 @@ def get_cluster_status(self, cluster_name: str) -> ClusterStatus:
160160 else :
161161 state = ClusterState .UNKNOWN
162162 else :
163- # REST API - use /slurm/v0.0.38 /partitions or similar
163+ # REST API - use /slurm/v0.0.39 /partitions or similar
164164 try :
165- result = self ._rest_request ("GET" , "/slurm/v0.0.38 /partitions" )
165+ result = self ._rest_request ("GET" , "/slurm/v0.0.39 /partitions" )
166166 state = ClusterState .UP if result else ClusterState .UNKNOWN
167167 except Exception :
168168 state = ClusterState .UNKNOWN
@@ -227,7 +227,7 @@ def get_cluster_resources(self, cluster_name: str) -> ResourceInfo:
227227 else :
228228 # REST API
229229 try :
230- result = self ._rest_request ("GET" , "/slurm/v0.0.38 /nodes" )
230+ result = self ._rest_request ("GET" , "/slurm/v0.0.39 /nodes" )
231231 # Parse REST API response
232232 gpus = []
233233 cpus = None
@@ -279,14 +279,14 @@ def submit_job(self, cluster_name: str, job_config: JobConfig) -> Dict[str, Any]
279279 job_id = line .split ()[- 1 ]
280280 break
281281 else :
282- # REST API - POST to /slurm/v0.0.38 /job/submit
282+ # REST API - POST to /slurm/v0.0.39 /job/submit
283283 data = {
284284 "script" : script_content ,
285285 "job" : {
286286 "name" : job_config .job_name ,
287287 },
288288 }
289- result = self ._rest_request ("POST" , "/slurm/v0.0.38 /job/submit" , data = data )
289+ result = self ._rest_request ("POST" , "/slurm/v0.0.39 /job/submit" , data = data )
290290 job_id = result .get ("job_id" )
291291
292292 return {"job_id" : job_id , "cluster_name" : cluster_name }
@@ -310,7 +310,7 @@ def get_job_logs(
310310 return output
311311 else :
312312 # REST API
313- result = self ._rest_request ("GET" , f"/slurm/v0.0.38 /job/{ job_id } " )
313+ result = self ._rest_request ("GET" , f"/slurm/v0.0.39 /job/{ job_id } " )
314314 return result .get ("logs" , str (result ))
315315
316316 def cancel_job (self , cluster_name : str , job_id : Union [str , int ]) -> Dict [str , Any ]:
@@ -321,7 +321,7 @@ def cancel_job(self, cluster_name: str, job_id: Union[str, int]) -> Dict[str, An
321321 return {"job_id" : job_id , "status" : "cancelled" }
322322 else :
323323 # REST API
324- result = self ._rest_request ("DELETE" , f"/slurm/v0.0.38 /job/{ job_id } " )
324+ result = self ._rest_request ("DELETE" , f"/slurm/v0.0.39 /job/{ job_id } " )
325325 return result
326326
327327 def list_jobs (self , cluster_name : str ) -> List [JobInfo ]:
@@ -355,7 +355,7 @@ def list_jobs(self, cluster_name: str) -> List[JobInfo]:
355355 return jobs
356356 else :
357357 # REST API
358- result = self ._rest_request ("GET" , "/slurm/v0.0.38 /jobs" )
358+ result = self ._rest_request ("GET" , "/slurm/v0.0.39 /jobs" )
359359 jobs = []
360360 for job_data in result .get ("jobs" , []):
361361 state_str = job_data .get ("job_state" , "UNKNOWN" ).upper ()
0 commit comments