Add fix

ray-project · Jul 12, 2022 · 9534c71 · 9534c71
1 parent 34259f5
commit 9534c71
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/python/ray/serve/deployment_function_node.py b/python/ray/serve/deployment_function_node.py
@@ -29,6 +29,11 @@ def __init__(
             func_options,
             other_args_to_resolve=other_args_to_resolve,
         )
+        version = (
+            self._bound_other_args_to_resolve["version"]
+            if "version" in self._bound_other_args_to_resolve
+            else None
+        )
         if "deployment_schema" in self._bound_other_args_to_resolve:
             deployment_schema: DeploymentSchema = self._bound_other_args_to_resolve[
                 "deployment_schema"
@@ -58,6 +63,7 @@ def __init__(
                 init_args=(),
                 init_kwargs={},
                 route_prefix=route_prefix,
+                version=version,
             )
         else:
             self._deployment: Deployment = Deployment(
@@ -68,6 +74,7 @@ def __init__(
                 init_kwargs=dict(),
                 ray_actor_options=func_options,
                 _internal=True,
+                version=version,
             )
         # TODO (jiaodong): Polish with async handle support later
         self._deployment_handle = RayServeLazySyncHandle(self._deployment.name)