Add ground_truth API and refactor

2022-06-19 17:15:52 +02:00 · 2022-06-19 17:15:52 +02:00 · 421f9bb726
commit 421f9bb726
parent c5cafbee47
1 changed files with 409 additions and 211 deletions
--- a/great_ai/src/great_ai/great_ai/output_models/classification_output.py
+++ b/great_ai/src/great_ai/great_ai/output_models/classification_output.py
@ -1,12 +1,9 @@
 from typing import Any, Optional, Union

-from pydantic import BaseModel
+from ..helper import HashableBaseModel


-class ClassificationOutput(BaseModel):
+class ClassificationOutput(HashableBaseModel):
    label: Union[str, int]
    confidence: float
    explanation: Optional[Any]
-
-    def __hash__(self) -> int:
-        return hash((type(self),) + tuple(self.__dict__.values()))