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/regression_output.py
+++ b/great_ai/src/great_ai/great_ai/output_models/regression_output.py
@ -1,11 +1,8 @@
 from typing import Any, Optional, Union

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


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