run-snapshot-async (#1330)

* add task_id to snapshot links --------- Co-authored-by: mike0sv <[email protected]>
evidentlyai · Oct 11, 2024 · 1a3728d · 1a3728d
1 parent 42d836a
commit 1a3728d
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/src/evidently/suite/base_suite.py b/src/evidently/suite/base_suite.py
@@ -500,6 +500,7 @@ def __iter__(self) -> Iterator[Tuple[str, str, DatasetID]]:
 class SnapshotLinks(BaseModel):
     datasets: DatasetInputOutputLinks = DatasetInputOutputLinks()
     computation_config_id: Optional[ComputationConfigID] = None
+    task_id: Optional[str] = None
 
 
 class Snapshot(BaseModel):

diff --git a/src/evidently/ui/config.py b/src/evidently/ui/config.py
@@ -117,8 +117,13 @@ def load_config(config_type: Type[TConfig], box: dict) -> TConfig:
             continue
         if section in ("renamed_vars", "dict_itemiterator"):
             continue
-        if section in config_type.__fields__:
-            component = parse_obj_as(config_type.__fields__[section].type_, component_dict)
+        if section == "additional_components":
+            for subsection, compoennt_subdict in component_dict.items():
+                component = parse_obj_as(SECTION_COMPONENT_TYPE_MAPPING.get(subsection, Component), compoennt_subdict)
+                components[subsection] = component
+        elif section in config_type.__fields__:
+            type_ = config_type.__fields__[section].type_
+            component = parse_obj_as(type_, component_dict)
             named_components[section] = component
         elif section in SECTION_COMPONENT_TYPE_MAPPING:
             component = parse_obj_as(SECTION_COMPONENT_TYPE_MAPPING[section], component_dict)