revert plugin.go

Signed-off-by: Kevin Su <pingsutw@apache.org>

go/tasks/plugins/webapi/agent/plugin.go

@@ -6,22 +6,21 @@ import (
 	"encoding/gob"
 	"fmt"
 
+	"github.com/flyteorg/flyteidl/gen/pb-go/flyteidl/admin"
+	"github.com/flyteorg/flytestdlib/config"
 	"google.golang.org/grpc/credentials"
 	"google.golang.org/grpc/credentials/insecure"
+
 	"google.golang.org/grpc/grpclog"
 
-	"github.com/flyteorg/flyteidl/gen/pb-go/flyteidl/admin"
 	flyteIdl "github.com/flyteorg/flyteidl/gen/pb-go/flyteidl/core"
 	"github.com/flyteorg/flyteidl/gen/pb-go/flyteidl/service"
 	pluginErrors "github.com/flyteorg/flyteplugins/go/tasks/errors"
 	"github.com/flyteorg/flyteplugins/go/tasks/pluginmachinery"
 	"github.com/flyteorg/flyteplugins/go/tasks/pluginmachinery/core"
 	"github.com/flyteorg/flyteplugins/go/tasks/pluginmachinery/core/template"
-	"github.com/flyteorg/flyteplugins/go/tasks/pluginmachinery/io"
 	"github.com/flyteorg/flyteplugins/go/tasks/pluginmachinery/ioutils"
 	"github.com/flyteorg/flyteplugins/go/tasks/pluginmachinery/webapi"
-	"github.com/flyteorg/flytestdlib/config"
-	"github.com/flyteorg/flytestdlib/logger"
 	"github.com/flyteorg/flytestdlib/promutils"
 	"google.golang.org/grpc"
 )
@@ -170,17 +169,11 @@ func (p Plugin) Status(ctx context.Context, taskCtx webapi.StatusContext) (phase
 	case admin.State_RETRYABLE_FAILURE:
 		return core.PhaseInfoRetryableFailure(pluginErrors.TaskFailedWithError, "failed to run the job", taskInfo), nil
 	case admin.State_SUCCEEDED:
-		var opReader io.OutputReader
 		if resource.Outputs != nil {
-			logger.Infof(ctx, "Agent returned an output")
-			opReader = ioutils.NewInMemoryOutputReader(resource.Outputs, nil, nil)
-		} else {
-			logger.Infof(ctx, "Agent didn't return any output, assuming file based outputs.")
-			opReader = ioutils.NewRemoteFileOutputReader(ctx, taskCtx.DataStore(), taskCtx.OutputWriter(), taskCtx.MaxDatasetSizeBytes())
-		}
-		err := taskCtx.OutputWriter().Put(ctx, opReader)
-		if err != nil {
-			return core.PhaseInfoUndefined, err
+			err := taskCtx.OutputWriter().Put(ctx, ioutils.NewInMemoryOutputReader(resource.Outputs, nil, nil))
+			if err != nil {
+				return core.PhaseInfoUndefined, err
+			}
 		}
 		return core.PhaseInfoSuccess(taskInfo), nil
 	}