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

Signed-off-by: Felix Ruess <[email protected]> Signed-off-by: Felix Ruess <[email protected]>
eapolinario · Jan 10, 2023 · 6297c94 · 6297c94
1 parent fe4efa0
commit 6297c94
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/flyteplugins/go/tasks/plugins/k8s/pod/container.go b/flyteplugins/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/flyteplugins/go/tasks/plugins/k8s/pod/plugin.go b/flyteplugins/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,