update numpy-flight (#93)

* Update Dockerfile

* cloud build

* fixing docker release

* Update README.md

* Update README.md

* Update cloudbuild.yaml

* fmt

* Update server.py

* cloudbuild with logging

* align server with example

* cloudbuild with logging

* Simpler server

* pyarrow dependency

* Simpler server

* Simpler server

* mini server

* explicit publication of image

* updating the server

* bringing back the full server

* passing all tests?

* logging

* passing all tests?

* server back on Ball problem

* remove pyarrow

* server back on Ball problem

* Makefile pointing to demo

* update of numpy-flight

Author: tschm

pyproject.toml

@@ -9,7 +9,7 @@ dependencies = [
     "cvxpy-base>=1.6.0",
     "numpy>=2",
     "clarabel>=0.10.0",
-    "numpy-flight>=0.0.7",
+    "numpy-flight>=0.0.9",
 ]
 
 [project.urls]
@@ -50,6 +50,7 @@ DEP003 = ["cvx"]
 
 [tool.bandit]
 exclude_dirs = ["tests"]
+exclude = ["B104"]
 
 [project.scripts]
 #server = "cvx.ball.server:BallServer.start"

src/cvx/ball/__init__.py

@@ -1,13 +0,0 @@
-#    Copyright 2025 Stanford University Convex Optimization Group
-#
-#    Licensed under the Apache License, Version 2.0 (the "License");
-#    you may not use this file except in compliance with the License.
-#    You may obtain a copy of the License at
-#
-#        http://www.apache.org/licenses/LICENSE-2.0
-#
-#    Unless required by applicable law or agreed to in writing, software
-#    distributed under the License is distributed on an "AS IS" BASIS,
-#    WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-#    See the License for the specific language governing permissions and
-#    limitations under the License.

src/cvx/ball/server.py

@@ -1,13 +1,8 @@
-import logging
-
 import numpy as np
-from np.flight import Server
+from tschm.flight import Server
 
 from cvx.ball.solver import min_circle_cvx
 
-logging.basicConfig(level=logging.INFO)
-logger = logging.getLogger(__name__)
-
 
 class BallServer(Server):
     def f(self, matrices: dict[str, np.ndarray]) -> dict[str, np.ndarray]:
@@ -28,72 +23,5 @@ def f(self, matrices: dict[str, np.ndarray]) -> dict[str, np.ndarray]:
         return {"radius": radius, "midpoint": midpoint, "points": matrix}
 
 
-# def serve(port=8080):
-#     # Create the server instance
-#     location = f"grpc://0.0.0.0:{port}"
-#     server = FlightServer(location=location)
-#
-#     # Start the server
-#     server.serve()
-#     print(f"Flight Server is listening on port {port}...")
-#
-#
-# # entry point for Docker
-# if __name__ == "__main__":  # pragma: no cover
-#     serve()
-#     # BallServer.start(host="0.0.0.0", port=8080)  # nosec: B104
-
-#
-# class FlightServer(pyarrow.flight.FlightServerBase):
-#     def __init__(self, location):
-#         super().__init__(location)
-#         logger.info(f"Flight Server started at {location}")
-#
-#     def do_get(self, context, ticket):
-#         try:
-#             logger.info(f"Received do_get request for ticket: {ticket}")
-#             # Example data
-#             data = pa.array([1, 2, 3, 4, 5])
-#             table = pa.table({"numbers": data})
-#             return pyarrow.flight.RecordBatchStream(table)
-#         except Exception as e:
-#             logger.error(f"Error in do_get: {e}")
-#             raise
-#
-#     def do_put(self, context, descriptor, reader):
-#         try:
-#             logger.info(f"Received do_put request for descriptor: {descriptor}")
-#             # Read the data (for demonstration, we just log the schema)
-#             table = reader.read_all()
-#             logger.info(f"Received table with schema: {table.schema}")
-#             return None
-#         except Exception as e:
-#             logger.error(f"Error in do_put: {e}")
-#             raise
-#
-#     def get_flight_info(self, context, descriptor):
-#         """Provide metadata about available flights."""
-#         key = descriptor.command
-#         if key not in self.flights:
-#             raise pyarrow.flight.FlightUnavailableError(f"Flight {key} not found")
-#
-#         # Create a FlightInfo object with metadata
-#         schema = self.flights[key].schema
-#         endpoint = pyarrow.flight.FlightEndpoint(key, [self.location])
-#         return pyarrow.flight.FlightInfo(schema, descriptor, [endpoint])
-
-
-def serve(port=8080):
-    # Create the server instance
-    # location = f"grpc://0.0.0.0:{port}"
-    server = BallServer(host="0.0.0.0", port=port)  # nosec:B104 # .start()
-    # FlightServer(location=location)
-
-    # Start the server
-    logger.info(f"Starting Flight Server on port {port}...")
-    # not needed
-    server.serve()
-
-
 if __name__ == "__main__":  # pragma: no cover
-    serve()
+    BallServer.start()