From 39020ca36a0e661040dfe51cc4c8f375771ad227 Mon Sep 17 00:00:00 2001 From: Xi Yan Date: Wed, 12 Mar 2025 01:30:19 -0700 Subject: [PATCH] post training job api --- llama_stack/apis/post_training/post_training.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/llama_stack/apis/post_training/post_training.py b/llama_stack/apis/post_training/post_training.py index 58e84eeee..3e7ab82fb 100644 --- a/llama_stack/apis/post_training/post_training.py +++ b/llama_stack/apis/post_training/post_training.py @@ -88,9 +88,7 @@ class QATFinetuningConfig(BaseModel): AlgorithmConfig = register_schema( - Annotated[ - Union[LoraFinetuningConfig, QATFinetuningConfig], Field(discriminator="type") - ], + Annotated[Union[LoraFinetuningConfig, QATFinetuningConfig], Field(discriminator="type")], name="AlgorithmConfig", )