added 'python-task' to registered task types (#290)

Signed-off-by: Daniel Rammer <[email protected]> Signed-off-by: Daniel Rammer <[email protected]>
flyteorg · Oct 21, 2022 · 932e97b · 932e97b
1 parent eed7326
commit 932e97b
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/go/tasks/plugins/k8s/pod/container.go b/go/tasks/plugins/k8s/pod/container.go
@@ -14,6 +14,7 @@ import (
 
 const (
 	ContainerTaskType = "container"
+	PythonTaskType    = "python-task"
 )
 
 type containerPodBuilder struct {

diff --git a/go/tasks/plugins/k8s/pod/plugin.go b/go/tasks/plugins/k8s/pod/plugin.go
@@ -157,7 +157,7 @@ func init() {
 	pluginmachinery.PluginRegistry().RegisterK8sPlugin(
 		k8s.PluginEntry{
 			ID:                  ContainerTaskType,
-			RegisteredTaskTypes: []pluginsCore.TaskType{ContainerTaskType},
+			RegisteredTaskTypes: []pluginsCore.TaskType{ContainerTaskType, PythonTaskType},
 			ResourceToWatch:     &v1.Pod{},
 			Plugin:              DefaultPodPlugin,
 			IsDefault:           true,
@@ -176,7 +176,7 @@ func init() {
 	pluginmachinery.PluginRegistry().RegisterK8sPlugin(
 		k8s.PluginEntry{
 			ID:                  podTaskType,
-			RegisteredTaskTypes: []pluginsCore.TaskType{ContainerTaskType, SidecarTaskType},
+			RegisteredTaskTypes: []pluginsCore.TaskType{ContainerTaskType, PythonTaskType, SidecarTaskType},
 			ResourceToWatch:     &v1.Pod{},
 			Plugin:              DefaultPodPlugin,
 			IsDefault:           true,