Add exception logging

2022-05-26 21:00:14 +02:00 · 2022-05-26 21:00:14 +02:00 · cdc43f75ac
commit cdc43f75ac
parent 47e8d8b33a
1 changed files with 144 additions and 182 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
@ -0,0 +1,11 @@
+from typing import Any, Optional, Union
+
+from pydantic import BaseModel
+
+
+class RegressionOutput(BaseModel):
+    value: Union[int, float]
+    explanation: Optional[Any]
+
+    def __hash__(self) -> int:
+        return hash((type(self),) + tuple(self.__dict__.values()))