Fix how ShellTask retrieves the Pod class name (#1132)

* Fix how ShellTask retrieves the Pod class name Signed-off-by: Matheus Moreno <[email protected]> * Set Pod class name as a constant Signed-off-by: Matheus Moreno <[email protected]> * Revert last commit Signed-off-by: Matheus Moreno <[email protected]> * Execute automatic linting Signed-off-by: Matheus Moreno <[email protected]> Signed-off-by: Matheus Moreno <[email protected]>
flyteorg · Aug 16, 2022 · dbe485d · dbe485d
1 parent 994a8bf
commit dbe485d
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/flytekit/extras/tasks/shell.py b/flytekit/extras/tasks/shell.py
@@ -132,7 +132,8 @@ def __init__(
             script_file = os.path.abspath(script_file)
 
         if task_config is not None:
-            if str(type(task_config)) != "flytekitplugins.pod.task.Pod":
+            fully_qualified_class_name = task_config.__module__ + "." + task_config.__class__.__name__
+            if not fully_qualified_class_name == "flytekitplugins.pod.task.Pod":
                 raise ValueError("TaskConfig can either be empty - indicating simple container task or a PodConfig.")
 
         # Each instance of NotebookTask instantiates an underlying task with a dummy function that will only be used