From 870271d51f520127748e5f49f0f386b552be7d86 Mon Sep 17 00:00:00 2001
From: Simon Mo <xmo@berkeley.edu>
Date: Thu, 9 Apr 2020 11:46:15 -0700
Subject: [PATCH] [Serve] Call serve.init in function handler (#7947)

---
 python/ray/serve/task_runner.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/python/ray/serve/task_runner.py b/python/ray/serve/task_runner.py
index ecc283fbb..8da8374b9 100644
--- a/python/ray/serve/task_runner.py
+++ b/python/ray/serve/task_runner.py
@@ -3,6 +3,7 @@ import traceback
 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.