Spaces:
Runtime error
Runtime error
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 | |