[Serve] Call serve.init in function handler (ray-project#7947)

hellcoderz · Apr 9, 2020 · 870271d · 870271d
1 parent d2b5c17
commit 870271d
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/python/ray/serve/task_runner.py b/python/ray/serve/task_runner.py
@@ -3,6 +3,7 @@
 import inspect
 
 import ray
+from ray import serve
 from ray.serve import context as serve_context
 from ray.serve.context import FakeFlaskRequest
 from collections import defaultdict
@@ -19,6 +20,8 @@ class TaskRunner:
     """
 
     def __init__(self, func_to_run):
+        serve.init()
+
         self.func = func_to_run
 
         # This parameter let argument inspection work with inner function.