move prometheus-client import

flyteorg · Sep 20, 2023 · e65f6c2 · e65f6c2
1 parent 054c099
commit e65f6c2
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/flytekit/clis/sdk_in_container/serve.py b/flytekit/clis/sdk_in_container/serve.py
@@ -4,8 +4,6 @@
 from flyteidl.service.agent_pb2_grpc import add_AsyncAgentServiceServicer_to_server
 from grpc import aio
 
-from flytekit.extend.backend.agent_service import AsyncAgentService
-
 _serve_help = """Start a grpc server for the agent service."""
 
 
@@ -47,6 +45,8 @@ async def _start_grpc_server(port: int, worker: int, timeout: int):
     try:
         from prometheus_client import start_http_server
 
+        from flytekit.extend.backend.agent_service import AsyncAgentService
+
         start_http_server(9090)
     except ImportError as e:
         click.secho(f"Failed to start the prometheus server with error {e}", fg="red")

diff --git a/setup.py b/setup.py
@@ -75,7 +75,6 @@
         "rich",
         "rich_click",
         "jsonpickle",
-        "prometheus-client",
     ],
     extras_require=extras_require,
     scripts=[