mirror of
https://github.com/wassname/ray.git
synced 2026-08-06 13:31:10 +08:00
[Tune] Fix tune serve integration example (#13233)
This commit is contained in:
@@ -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)
|
||||
|
||||
|
||||
@@ -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
|
||||
|
||||
|
||||
Reference in New Issue
Block a user