Update plugins/flytekit-kf-pytorch/flytekitplugins/kfpytorch/task.py

Signed-off-by: Fabio Graetz <[email protected]>
flyteorg · Sep 18, 2023 · 027feb0 · 027feb0
1 parent 7f6fd68
commit 027feb0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/plugins/flytekit-kf-pytorch/flytekitplugins/kfpytorch/task.py b/plugins/flytekit-kf-pytorch/flytekitplugins/kfpytorch/task.py
@@ -213,7 +213,7 @@ class ElasticWorkerResult(NamedTuple):
     """
 
     return_value: Any
-    decks: list[flytekit.Deck]
+    decks: List[flytekit.Deck]
 
 
 def spawn_helper(