From ce3b1ee44136f950a6fa678b88adeb4a587937a2 Mon Sep 17 00:00:00 2001 From: Jusong Yu Date: Mon, 22 Jul 2024 11:08:12 +0200 Subject: [PATCH] Cast job_id to str to conform with type of _jobs_cache fixes #6542 --- src/aiida/engine/processes/calcjobs/manager.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/aiida/engine/processes/calcjobs/manager.py b/src/aiida/engine/processes/calcjobs/manager.py index 75b9bb6ab1..4a0bb3a4f7 100644 --- a/src/aiida/engine/processes/calcjobs/manager.py +++ b/src/aiida/engine/processes/calcjobs/manager.py @@ -114,7 +114,7 @@ async def _get_jobs_from_scheduler(self) -> Dict[Hashable, 'JobInfo']: self.logger.info(f'AuthInfo<{self._authinfo.pk}>: successfully retrieved status of active jobs') for job_id, job_info in scheduler_response.items(): - jobs_cache[job_id] = job_info + jobs_cache[str(job_id)] = job_info return jobs_cache @@ -146,7 +146,7 @@ async def _update_job_info(self) -> None: else: for job_id, future in self._job_update_requests.items(): if not future.done(): - future.set_result(self._jobs_cache.get(job_id, None)) + future.set_result(self._jobs_cache.get(str(job_id), None)) finally: self._job_update_requests = {}