phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-08-03 17:29:01 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

remove more imports

Browse source
This commit is contained in:
Xi Yan 2024-12-26 16:43:30 -08:00
parent 93ed8aa814
commit bb0a3f5c8e
3 changed files with 5 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

8
llama_stack/apis/post_training/post_training.py
View file

@ -7,17 +7,17 @@
from datetime import datetime from datetime import datetime
from enum import Enum 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 llama_models.schema_utils import json_schema_type, webmethod
from pydantic import BaseModel, Field from pydantic import BaseModel, Field
from typing_extensions import Annotated 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.common.job_types import JobStatus
from llama_stack.apis.datasets import * # noqa: F403 from llama_stack.apis.common.training_types import Checkpoint
from llama_stack.apis.common.training_types import * # noqa: F403
@json_schema_type @json_schema_type

1
llama_stack/apis/scoring/scoring.py
View file

@ -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 llama_models.schema_utils import json_schema_type, webmethod
from pydantic import BaseModel from pydantic import BaseModel
# from llama_models.llama3.api.datatypes import * # noqa: F403
from llama_stack.apis.scoring_functions import ScoringFn, ScoringFnParams from llama_stack.apis.scoring_functions import ScoringFn, ScoringFnParams

3
llama_stack/apis/synthetic_data_generation/synthetic_data_generation.py
View file

@ -6,13 +6,12 @@
from enum import Enum 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 llama_models.schema_utils import json_schema_type, webmethod
from pydantic import BaseModel from pydantic import BaseModel
from llama_models.llama3.api.datatypes import * # noqa: F403
from llama_stack.apis.inference import Message from llama_stack.apis.inference import Message