From bb0a3f5c8e729ff7c9448ca7fd64a9a8bd2d2827 Mon Sep 17 00:00:00 2001 From: Xi Yan Date: Thu, 26 Dec 2024 16:43:30 -0800 Subject: [PATCH] remove more imports --- llama_stack/apis/post_training/post_training.py | 8 ++++---- llama_stack/apis/scoring/scoring.py | 1 - .../synthetic_data_generation.py | 3 +-- 3 files changed, 5 insertions(+), 7 deletions(-) diff --git a/llama_stack/apis/post_training/post_training.py b/llama_stack/apis/post_training/post_training.py index fdbaa364d..1c2d2d6e2 100644 --- a/llama_stack/apis/post_training/post_training.py +++ b/llama_stack/apis/post_training/post_training.py @@ -7,17 +7,17 @@ from datetime import datetime from enum import Enum -from typing import Any, Dict, List, Optional, Protocol, Union +from typing import Any, Dict, List, Literal, Optional, Protocol, Union from llama_models.schema_utils import json_schema_type, webmethod from pydantic import BaseModel, Field from typing_extensions import Annotated -from llama_models.llama3.api.datatypes import * # noqa: F403 +from llama_stack.apis.common.content_types import URL + from llama_stack.apis.common.job_types import JobStatus -from llama_stack.apis.datasets import * # noqa: F403 -from llama_stack.apis.common.training_types import * # noqa: F403 +from llama_stack.apis.common.training_types import Checkpoint @json_schema_type diff --git a/llama_stack/apis/scoring/scoring.py b/llama_stack/apis/scoring/scoring.py index 17d1426b5..453e35f6d 100644 --- a/llama_stack/apis/scoring/scoring.py +++ b/llama_stack/apis/scoring/scoring.py @@ -9,7 +9,6 @@ from typing import Any, Dict, List, Optional, Protocol, runtime_checkable from llama_models.schema_utils import json_schema_type, webmethod from pydantic import BaseModel -# from llama_models.llama3.api.datatypes import * # noqa: F403 from llama_stack.apis.scoring_functions import ScoringFn, ScoringFnParams diff --git a/llama_stack/apis/synthetic_data_generation/synthetic_data_generation.py b/llama_stack/apis/synthetic_data_generation/synthetic_data_generation.py index 4ffaa4d1e..13b209912 100644 --- a/llama_stack/apis/synthetic_data_generation/synthetic_data_generation.py +++ b/llama_stack/apis/synthetic_data_generation/synthetic_data_generation.py @@ -6,13 +6,12 @@ from enum import Enum -from typing import Any, Dict, List, Optional, Protocol +from typing import Any, Dict, List, Optional, Protocol, Union from llama_models.schema_utils import json_schema_type, webmethod from pydantic import BaseModel -from llama_models.llama3.api.datatypes import * # noqa: F403 from llama_stack.apis.inference import Message