The gRPC server now has the port option that allows to specify the se…

…rver's listening port. (#1677)

wandb/cli/cli.py

@@ -242,19 +242,20 @@ def login(key, host, cloud, relogin, anonymously, no_offline=False):
 @cli.command(
     context_settings=CONTEXT, help="Run a grpc server", name="grpc-server", hidden=True
 )
+@click.option("--port", "-p", default=None, help="The port to use.")
 @display_error
-def grpc_server(project=None, entity=None):
+def grpc_server(project=None, entity=None, port=None):
     _ = util.get_module(
         "grpc",
         required="grpc-server requires the grpcio library, run pip install wandb[grpc]",
     )
     from wandb.server.grpc_server import main as grpc_server
 
-    grpc_server()
+    grpc_server(port=port)
 
 
 @cli.command(context_settings=CONTEXT, help="Run a SUPER agent", hidden=True)
-@click.option("--project", "-p", default=None, help="The project use.")
+@click.option("--project", "-p", default=None, help="The project to use.")
 @click.option("--entity", "-e", default=None, help="The entity to use.")
 @click.argument("agent_spec", nargs=-1)
 @display_error

wandb/server/grpc_server.py

@@ -164,13 +164,13 @@ def cleanup(self):
         # No printing allowed from here until redirect restore!!!
 
 
-def serve(backend):
+def serve(backend, port):
     try:
         server = grpc.server(futures.ThreadPoolExecutor(max_workers=10))
         wandb_server_pb2_grpc.add_InternalServiceServicer_to_server(
             InternalServiceServicer(server, backend), server
         )
-        server.add_insecure_port("[::]:50051")
+        server.add_insecure_port(f"[::]:{port}")
         server.start()
         server.wait_for_termination()
         # print("server shutting down")
@@ -179,12 +179,12 @@ def serve(backend):
         print("control-c")
 
 
-def main():
+def main(port=50051):
     try:
         logging.basicConfig()
         backend = Backend()
         backend.setup()
-        serve(backend)
+        serve(backend, port if port is not None else 50051)
     except KeyboardInterrupt:
         print("outer control-c")