From 80070b966a993fb18749395008c976a824cf65d7 Mon Sep 17 00:00:00 2001 From: Ihar Hrachyshka Date: Tue, 18 Feb 2025 09:27:21 -0500 Subject: [PATCH] refactor: simplify job status extraction a bit Signed-off-by: Ihar Hrachyshka --- .../providers/inline/post_training/torchtune/post_training.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/llama_stack/providers/inline/post_training/torchtune/post_training.py b/llama_stack/providers/inline/post_training/torchtune/post_training.py index c77d9305f..d13347240 100644 --- a/llama_stack/providers/inline/post_training/torchtune/post_training.py +++ b/llama_stack/providers/inline/post_training/torchtune/post_training.py @@ -119,9 +119,7 @@ class TorchtunePostTrainingImpl: @webmethod(route="/post-training/job/status") async def get_training_job_status(self, job_uuid: str) -> Optional[PostTrainingJobStatusResponse]: - if job_uuid in self.jobs_status: - return self.jobs_status[job_uuid] - return None + return self.jobs_status.get(job_uuid, None) @webmethod(route="/post-training/job/cancel") async def cancel_training_job(self, job_uuid: str) -> None: