nit

Signed-off-by: Kevin Su <[email protected]>
flyteorg · Feb 20, 2024 · a46196d · a46196d
1 parent 779b336
commit a46196d
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/flyteplugins/go/tasks/plugins/webapi/agent/client.go b/flyteplugins/go/tasks/plugins/webapi/agent/client.go
@@ -136,6 +136,12 @@ func initializeAgentRegistry(cs *ClientSet) (Registry, error) {
 		}
 
 		for _, agent := range res.GetAgents() {
+			deprecatedSupportedTaskTypes := agent.DeprecatedSupportedTaskTypes
+			for _, supportedTaskType := range deprecatedSupportedTaskTypes {
+				agent := &Agent{AgentDeployment: agentDeployment, IsSync: agent.IsSync}
+				agentRegistry[supportedTaskType] = map[int32]*Agent{defaultTaskTypeVersion: agent}
+			}
+
 			supportedTaskTypes := agent.SupportedTaskTypes
 			for _, supportedTaskType := range supportedTaskTypes {
 				agent := &Agent{AgentDeployment: agentDeployment, IsSync: agent.IsSync}