[Tune] Fix tune serve integration example (#13233)

This commit is contained in:
Kai Fricke
2021-01-06 17:02:04 +01:00
committed by GitHub
parent 32dc5676b4
commit 97211a6170
2 changed files with 4 additions and 3 deletions
@@ -440,7 +440,7 @@ class MNISTBackend:
#######################################################################
# We would like to have a fixed location where we store the currently
# active model. We call this directory ``model_dir``. Everytime we
# active model. We call this directory ``model_dir``. Every time we
# would like to update our model, we copy the checkpoint of the new
# model to this directory. We then create a new backend pointing to
# that directory, route all the traffic on our model endpoint to this
@@ -615,7 +615,7 @@ if __name__ == "__main__":
args = parser.parse_args()
if args.smoke_test:
ray.init(num_cpus=2)
ray.init(num_cpus=3)
model_dir = os.path.expanduser(args.model_dir)
+2 -1
View File
@@ -7,7 +7,8 @@ import inspect
import threading
import time
import uuid
from collections import defaultdict, deque, Mapping, Sequence
from collections import defaultdict, deque
from collections.abc import Mapping, Sequence
from datetime import datetime
from threading import Thread