ob-metaflow 2.10.7.3__py2.py3-none-any.whl → 2.10.9.1__py2.py3-none-any.whl

metaflow/plugins/cards/card_modules/convert_to_native_type.py

@@ -44,7 +44,7 @@ def _full_classname(obj):
 
 
 class TaskToDict:
-    def __init__(self, only_repr=False):
+    def __init__(self, only_repr=False, runtime=False):
         # this dictionary holds all the supported functions
         import reprlib
         import pprint
@@ -59,6 +59,7 @@ class TaskToDict:
         r.maxlist = 100
         r.maxlevel = 3
         self._repr = r
+        self._runtime = runtime
         self._only_repr = only_repr
         self._supported_types = {
             "tuple": self._parse_tuple,
@@ -90,11 +91,16 @@ class TaskToDict:
             stderr=task.stderr,
             stdout=task.stdout,
             created_at=task.created_at.strftime(TIME_FORMAT),
-            finished_at=task.finished_at.strftime(TIME_FORMAT),
+            finished_at=None,
             pathspec=task.pathspec,
             graph=graph,
             data={},
         )
+        if not self._runtime:
+            if task.finished_at is not None:
+                task_dict.update(
+                    dict(finished_at=task.finished_at.strftime(TIME_FORMAT))
+                )
         task_dict["data"], type_infered_objects = self._create_task_data_dict(task)
         task_dict.update(type_infered_objects)
         return task_dict
@@ -310,7 +316,7 @@ class TaskToDict:
         time_format = "%Y-%m-%dT%H:%M:%S%Z"
         truncate_long_objects = (
             lambda x: x.astype("string").str.slice(0, 30) + "..."
-            if x.astype("string").str.len().max() > 30
+            if len(x) > 0 and x.astype("string").str.len().max() > 30
             else x.astype("string")
         )
         type_parser = {