diff --git a/flytepropeller/pkg/compiler/common/pretty_print_test.go b/flytepropeller/pkg/compiler/common/pretty_print_test.go index e8586ae2a5..1b646e40af 100644 --- a/flytepropeller/pkg/compiler/common/pretty_print_test.go +++ b/flytepropeller/pkg/compiler/common/pretty_print_test.go @@ -1,9 +1,10 @@ package common import ( - "google.golang.org/protobuf/types/known/structpb" "testing" + "google.golang.org/protobuf/types/known/structpb" + "github.com/stretchr/testify/assert" "github.com/flyteorg/flyte/flyteidl/gen/pb-go/flyteidl/core"