cortexgrid 0.3.18__tar.gz → 0.3.20__tar.gz

{cortexgrid-0.3.18 → cortexgrid-0.3.20}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: cortexgrid
-Version: 0.3.18
+Version: 0.3.20
 Summary: Connect your ML code to the RoboLab compute cluster — Ray, MLflow, and S3
 Project-URL: Homepage, https://github.com/robodatalab/cortexgrid
 Project-URL: Repository, https://github.com/robodatalab/cortexgrid

{cortexgrid-0.3.18 → cortexgrid-0.3.20}/cortexgrid/serve.py

@@ -8,8 +8,8 @@
         async def predict(self, xs: list[float]) -> list[float]: ...
 
 Unlike `ray.serve.ingress`, it builds the FastAPI app from the class's
-`serve.endpoint` methods, and the class is left unwrapped: the FastAPI app
-and the client generated for it are only recorded on it, and
+`serve.endpoint` methods, and the class is left unwrapped: the FastAPI app,
+its routes and the client generated for it are only recorded on it, and
 `cortexgrid._serve_entry.build` applies Ray's ingress when it builds the Serve
 application on the cluster.
 
@@ -59,6 +59,10 @@ def ingress(cls: _T) -> _T:
         if getattr(method, _ENDPOINT_ATTR, False):
             marshalling = _EndpointMarshalling.of(method)
             route = _route_of(method, marshalling)
+            route_name = f"__cortexgrid_{name}_route__"
+            route.__module__ = cls.__module__
+            route.__qualname__ = f"{cls.__qualname__}.{route_name}"
+            setattr(cls, route_name, route)
             app.add_api_route(f"/{name}", route, methods=["POST"])
             calls[name] = _call_of(name, method, marshalling)
     client = type(f"{cls.__name__}Client", (_EndpointsClient,), calls)
@@ -89,7 +93,7 @@ class _EndpointMarshalling:
         parameter_values = signature.parameters.values()
         parameters_in_order = list(parameter_values)
         signature_without_self = signature.replace(parameters=parameters_in_order[1:])
-        hints = get_type_hints(method)
+        hints = get_type_hints(method, include_extras=True)
         returned_hint = hints.pop("return")
         if inspect.isasyncgenfunction(method):
             streamed_hints = get_args(returned_hint)
@@ -157,7 +161,6 @@ def _route_of(
             return answered
 
     route.__name__ = method.__name__
-    route.__qualname__ = method.__qualname__
     return route
 
 

{cortexgrid-0.3.18 → cortexgrid-0.3.20}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "cortexgrid"
-version = "0.3.18"
+version = "0.3.20"
 description = "Connect your ML code to the RoboLab compute cluster — Ray, MLflow, and S3"
 readme = "docs/cortexgrid/README.md"
 license = "Apache-2.0"