diff --git a/backend/cluster.py b/backend/cluster.py index f2fe6e152cc5c920bb1a7378620ac900806bff85..eceaea98fd7420db07d54e3c6f02daf5d3d05152 100644 --- a/backend/cluster.py +++ b/backend/cluster.py @@ -114,6 +114,7 @@ def _fetch_and_parse_oarnodes_dict() -> tuple[dict, list[str]]: gpu_model_name: str = PRETTY_GPU_MODELS[core_dict['gputype']] nodes_dict[host] = { 'gpu_model': f"{gpu_model_name} ({core_dict['gpumem']}GB)", + 'state': core_dict['state'], 'gpus': {}, 'running_jobs': {} } @@ -206,8 +207,7 @@ def _fetch_and_parse_oarstat_dict(jobs_list: list[str]) -> dict: return oarstat_filtered_dict -def _merge(nodes_dict: dict, - jobs_dict: dict) -> dict: +def _merge(nodes_dict: dict, jobs_dict: dict) -> dict: output_dict: dict = {}