Skip to content
This repository was archived by the owner on Dec 16, 2025. It is now read-only.

Commit c04b4cf

Browse files
committed
change to use openapi 39
1 parent 885e067 commit c04b4cf

File tree

1 file changed

+8
-8
lines changed

1 file changed

+8
-8
lines changed

src/lattice/providers/slurm.py

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)