fixed imports

Signed-off-by: Daniel Rammer <daniel@union.ai>
flyteorg · Aug 22, 2023 · e92e374 · e92e374
1 parent 1f14040
commit e92e374
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/cmd/entrypoints/serve.go b/cmd/entrypoints/serve.go
@@ -7,6 +7,7 @@ import (
 	"github.com/flyteorg/datacatalog/pkg/rpc/datacatalogservice"
 	"github.com/flyteorg/datacatalog/pkg/runtime"
 	"github.com/flyteorg/flytestdlib/contextutils"
+	"github.com/flyteorg/flytestdlib/logger"
 	"github.com/flyteorg/flytestdlib/profutils"
 	"github.com/flyteorg/flytestdlib/promutils/labeled"
 

diff --git a/cmd/entrypoints/serve_dummy.go b/cmd/entrypoints/serve_dummy.go
@@ -5,6 +5,7 @@ import (
 
 	"github.com/flyteorg/datacatalog/pkg/config"
 	"github.com/flyteorg/datacatalog/pkg/rpc/datacatalogservice"
+	"github.com/flyteorg/flytestdlib/logger"
 	"github.com/spf13/cobra"
 )
 
@@ -17,7 +18,7 @@ var serveDummyCmd = &cobra.Command{
 
 		// serve a http healthcheck endpoint
 		go func() {
-			err := ServeHTTPHealthCheck(ctx, cfg)
+			err := datacatalogservice.ServeHTTPHealthCheck(ctx, cfg)
 			if err != nil {
 				logger.Errorf(ctx, "Unable to serve http", cfg.GetGrpcHostAddress(), err)
 			}