fixing location for index-lookup file for dynamic task (flyteorg#46)

eapolinario · Jan 17, 2020 · 21dd0e5 · 21dd0e5
1 parent 88298ee
commit 21dd0e5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/flyteplugins/go/tasks/plugins/array/catalog.go b/flyteplugins/go/tasks/plugins/array/catalog.go
@@ -110,7 +110,7 @@ func DetermineDiscoverability(ctx context.Context, tCtx core.TaskExecutionContex
 
 		indexLookup := CatalogBitsetToLiteralCollection(cachedResults, resp.GetResultsSize())
 		// TODO: Is the right thing to use?  Haytham please take a look
-		indexLookupPath, err := ioutils.GetIndexLookupPath(ctx, tCtx.DataStore(), tCtx.OutputWriter().GetOutputPrefixPath())
+		indexLookupPath, err := ioutils.GetIndexLookupPath(ctx, tCtx.DataStore(), tCtx.InputReader().GetInputPrefixPath())
 		if err != nil {
 			return state, err
 		}