Fixed register error

Signed-off-by: Kevin Su <[email protected]>
flyteorg · Sep 28, 2021 · 29b14f3 · 29b14f3
1 parent 79dd783
commit 29b14f3
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/flytekit/core/interface.py b/flytekit/core/interface.py
@@ -308,7 +308,7 @@ def transform_variable_map(
         for k, v in variable_map.items():
             res[k] = transform_type(v, descriptions.get(k, k))
             if isinstance(v, FlytePickle):
-                res[k].type.metadata = {"python_class_name": typing.cast(FlytePickle, v.python_type)}
+                res[k].type.metadata = {"python_class_name": typing.cast(FlytePickle, v.python_type.__name__)}
     return res
 
 

diff --git a/flytekit/types/pickle/pickle.py b/flytekit/types/pickle/pickle.py
@@ -15,7 +15,7 @@
 
 class FlytePickle(object):
     """
-    This type is only used by flytekit. User should not use this type.
+    This type is only used by flytekit internally. User should not use this type.
     Any type that flyte can't recognize will become FlytePickle
     """