Support Deferred Type Annotations in Dataset Previews (#1798)

* use include for previewType Signed-off-by: Sajid Alam <[email protected]> * update flowchart to accept name or annotation for preview type Signed-off-by: Sajid Alam <[email protected]> * revert FE changes Signed-off-by: Sajid Alam <[email protected]> --------- Signed-off-by: Sajid Alam <[email protected]>
kedro-org · Mar 12, 2024 · 7cf36bd · 7cf36bd
1 parent 1ff461d
commit 7cf36bd
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/package/kedro_viz/models/flowchart.py b/package/kedro_viz/models/flowchart.py
@@ -826,7 +826,15 @@ def set_preview_type(cls, _):
             return None
 
         try:
-            return inspect.signature(cls.dataset.preview).return_annotation.__name__
+            preview_type_annotation = inspect.signature(
+                cls.dataset.preview
+            ).return_annotation
+            # Attempt to get the name attribute, if it exists.
+            # Otherwise, use str to handle the annotation directly.
+            preview_type_name = getattr(
+                preview_type_annotation, "__name__", str(preview_type_annotation)
+            )
+            return preview_type_name
 
         except Exception as exc:  # pylint: disable=broad-except # pragma: no cover
             logger.warning(