Don't handle dynamic execution scope

Signed-off-by: Fabio Graetz <fabiograetz@googlemail.com>
flyteorg · Apr 27, 2023 · 27fcb29 · 27fcb29
1 parent 28ade30
commit 27fcb29
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/plugins/flytekit-kf-pytorch/flytekitplugins/kfpytorch/task.py b/plugins/flytekit-kf-pytorch/flytekitplugins/kfpytorch/task.py
@@ -217,10 +217,7 @@ def execute(self, **kwargs) -> Any:
         """
         from flytekit.exceptions import scopes as exception_scopes
 
-        if self.execution_mode == self.ExecutionBehavior.DEFAULT:
-            return exception_scopes.user_entry_point(self._execute)(**kwargs)
-        elif self.execution_mode == self.ExecutionBehavior.DYNAMIC:
-            return self.dynamic_execute(self._execute, **kwargs)
+        return exception_scopes.user_entry_point(self._execute)(**kwargs)
 
     def get_custom(self, settings: SerializationSettings) -> Optional[Dict[str, Any]]:
         if self.task_config.nnodes == 1: