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 c7d8d6758..d20e11b11 100644 --- a/llama_stack/providers/inline/post_training/torchtune/post_training.py +++ b/llama_stack/providers/inline/post_training/torchtune/post_training.py @@ -123,7 +123,8 @@ class TorchtunePostTrainingImpl: training_config: TrainingConfig, hyperparam_search_config: dict[str, Any], logger_config: dict[str, Any], - ) -> PostTrainingJob: ... + ) -> PostTrainingJob: + raise NotImplementedError() async def get_training_jobs(self) -> ListPostTrainingJobsResponse: return ListPostTrainingJobsResponse( diff --git a/pyproject.toml b/pyproject.toml index b41e03615..04a6a685e 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -256,7 +256,6 @@ exclude = [ "^llama_stack/providers/inline/inference/sentence_transformers/sentence_transformers\\.py$", "^llama_stack/providers/inline/inference/vllm/", "^llama_stack/providers/inline/post_training/common/validator\\.py$", - "^llama_stack/providers/inline/post_training/torchtune/post_training\\.py$", "^llama_stack/providers/inline/safety/code_scanner/", "^llama_stack/providers/inline/safety/llama_guard/", "^llama_stack/providers/inline/safety/prompt_guard/",