_type_as_str

evidentlyai · Aug 8, 2024 · bb07ee9 · bb07ee9
1 parent 3ec352b
commit bb07ee9
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/src/evidently/pydantic_utils.py b/src/evidently/pydantic_utils.py
@@ -178,7 +178,7 @@ def __get_classpath__(cls):
 
     def __init_subclass__(cls):
         super().__init_subclass__()
-        if cls == PolymorphicModel:
+        if cls == PolymorphicModel or vars(cls).get("_type_as_str"):
             return
         typename = cls.__get_type__()
         type_field = cls.__fields__["type"]

diff --git a/src/evidently/ui/dashboards/base.py b/src/evidently/ui/dashboards/base.py
@@ -5,6 +5,7 @@
 from functools import wraps
 from typing import TYPE_CHECKING
 from typing import Any
+from typing import ClassVar
 from typing import Dict
 from typing import List
 from typing import Optional
@@ -130,6 +131,8 @@ def inner(self: "DashboardPanel", *args, **kwargs) -> BaseWidgetInfo:
 
 
 class DashboardPanel(EnumValueMixin, PolymorphicModel):
+    _type_as_str: ClassVar[bool] = True
+
     id: PanelID = Field(default_factory=uuid.uuid4)
     title: str
     filter: ReportFilter