Spaces:
Running
on
Zero
Running
on
Zero
File size: 570 Bytes
df6c67d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 |
from typing import Literal, Optional, Union
from pydantic import BaseModel, Field
from inference.enterprise.workflows.entities.base import GraphNone
class InferenceImage(BaseModel, GraphNone):
type: Literal["InferenceImage"]
name: str
def get_type(self) -> str:
return self.type
class InferenceParameter(BaseModel, GraphNone):
type: Literal["InferenceParameter"]
name: str
default_value: Optional[Union[float, int, str, bool, list, set]] = Field(
default=None
)
def get_type(self) -> str:
return self.type
|