Move files

great_ai/views/__init__.py

@@ -6,6 +6,11 @@ from .function_metadata import FunctionMetadata
 from .health_check_response import HealthCheckResponse
 from .model import Model
 from .operators import operators
+from .outputs import (
+    ClassificationOutput,
+    MultiLabelClassificationOutput,
+    RegressionOutput,
+)
 from .query import Query
 from .route_config import RouteConfig
 from .sort_by import SortBy

great_ai/views/hashable_base_model.py

@@ -0,0 +1,6 @@
+from pydantic import BaseModel
+
+
+class HashableBaseModel(BaseModel):
+    def __hash__(self) -> int:
+        return hash((type(self),) + tuple(self.__dict__.values()))

great_ai/views/outputs/__init__.py

@@ -0,0 +1,3 @@
+from .classification_output import ClassificationOutput
+from .multi_label_classification_output import MultiLabelClassificationOutput
+from .regression_output import RegressionOutput

great_ai/views/outputs/classification_output.py

@@ -0,0 +1,9 @@
+from typing import Any, Optional, Union
+
+from ..hashable_base_model import HashableBaseModel
+
+
+class ClassificationOutput(HashableBaseModel):
+    label: Union[str, int]
+    confidence: float
+    explanation: Optional[Any]

great_ai/views/outputs/multi_label_classification_output.py

@@ -0,0 +1,8 @@
+from typing import List
+
+from ..hashable_base_model import HashableBaseModel
+from .classification_output import ClassificationOutput
+
+
+class MultiLabelClassificationOutput(HashableBaseModel):
+    labels: List[ClassificationOutput] = []

great_ai/views/outputs/regression_output.py

@@ -0,0 +1,8 @@
+from typing import Any, Optional, Union
+
+from ..hashable_base_model import HashableBaseModel
+
+
+class RegressionOutput(HashableBaseModel):
+    value: Union[int, float]
+    explanation: Optional[Any]

great_ai/views/outputs/sequence_labeling_output.py

@@ -0,0 +1 @@
+# todo

great_ai/views/trace.py

@@ -3,7 +3,7 @@ from typing import Any, Dict, Generic, List, Optional, TypeVar
 
 from pydantic import Extra
 
-from ..helper import HashableBaseModel
+from .hashable_base_model import HashableBaseModel
 from .model import Model
 
 T = TypeVar("T")