added raw-container to registered task types (#305)

Signed-off-by: Felix Ruess <[email protected]> Signed-off-by: Felix Ruess <[email protected]>
flyteorg · Jan 10, 2023 · 109224c · 109224c
1 parent 36c0b41
commit 109224c
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/go/tasks/plugins/k8s/pod/container.go b/go/tasks/plugins/k8s/pod/container.go
@@ -13,8 +13,9 @@ import (
 )
 
 const (
-	ContainerTaskType = "container"
-	PythonTaskType    = "python-task"
+	ContainerTaskType    = "container"
+	PythonTaskType       = "python-task"
+	RawContainerTaskType = "raw-container"
 )
 
 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, PythonTaskType},
+			RegisteredTaskTypes: []pluginsCore.TaskType{ContainerTaskType, PythonTaskType, RawContainerTaskType},
 			ResourceToWatch:     &v1.Pod{},
 			Plugin:              DefaultPodPlugin,
 			IsDefault:           true,
@@ -176,7 +176,7 @@ func init() {
 	pluginmachinery.PluginRegistry().RegisterK8sPlugin(
 		k8s.PluginEntry{
 			ID:                  podTaskType,
-			RegisteredTaskTypes: []pluginsCore.TaskType{ContainerTaskType, PythonTaskType, SidecarTaskType},
+			RegisteredTaskTypes: []pluginsCore.TaskType{ContainerTaskType, PythonTaskType, RawContainerTaskType, SidecarTaskType},
 			ResourceToWatch:     &v1.Pod{},
 			Plugin:              DefaultPodPlugin,
 			IsDefault:           true,