Fix with new plugins changes

flyteorg · Oct 7, 2019 · 0b49e4f · 0b49e4f
1 parent 6a143ea
commit 0b49e4f
Show file tree

Hide file tree

Showing 3 changed files with 35 additions and 29 deletions.
diff --git a/Gopkg.lock b/Gopkg.lock
diff --git a/pkg/controller/nodes/task/handler.go b/pkg/controller/nodes/task/handler.go
@@ -473,7 +473,7 @@ func (t Handler) Finalize(ctx context.Context, nCtx handler.NodeExecutionContext
 
 func New(_ context.Context, kubeClient executors.Client, client catalog.Client, scope promutils.Scope) *Handler {
 	// TODO NewShould take apointer
-	async, err := catalog.NewAsyncClient(client, *catalog.GetConfig())
+	async, err := catalog.NewAsyncClient(client, *catalog.GetConfig(), scope.NewSubScope("catalog"))
 	if err != nil {
 		return nil
 	}

diff --git a/pkg/controller/nodes/task/taskexec_context.go b/pkg/controller/nodes/task/taskexec_context.go
@@ -61,6 +61,12 @@ type taskExecutionContext struct {
 	c   pluginCatalog.AsyncClient
 }
 
+func (t *taskExecutionContext) EnqueueOwner() pluginCore.SignalOwner {
+	return func(ctx context.Context) {
+		t.NodeExecutionContext.EnqueueOwnerFunc()
+	}
+}
+
 func (t *taskExecutionContext) GetTaskRefreshIndicator() func() {
 	return func() {
 		_ = t.NodeExecutionContext.EnqueueOwnerFunc()