hiro/ray
1
0
Fork 0
mirror of https://github.com/vale981/ray synced 2025-03-06 10:31:39 -05:00
Code Issues Projects Releases Packages Wiki Activity Actions

[Serve] Simplify Validation (#8316)

Browse source
This commit is contained in:
Simon Mo 2020-05-04 20:31:23 -07:00 • committed by GitHub
parent e93ec3134a
commit ca929671b6
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
python/ray/serve/config.py
View file

@ -96,21 +96,21 @@ class ReplicaConfig:
"actor_init_args is not allowed.")
else:
num_cpus = self.ray_actor_options.get("num_cpus", 0)
if not (isinstance(num_cpus, int) or isinstance(num_cpus, float)):
if not isinstance(num_cpus, (int, float)):
raise TypeError(
"num_cpus in ray_actor_options must be an int or a float.")
elif num_cpus < 0:
raise ValueError("num_cpus in ray_actor_options must be >= 0.")
num_gpus = self.ray_actor_options.get("num_gpus", 0)
if not (isinstance(num_gpus, int) or isinstance(num_gpus, float)):
if not isinstance(num_gpus, (int, float)):
raise TypeError(
"num_gpus in ray_actor_options must be an int or a float.")
elif num_gpus < 0:
raise ValueError("num_gpus in ray_actor_options must be >= 0.")
memory = self.ray_actor_options.get("memory", 0)
if not (isinstance(memory, int) or isinstance(memory, float)):
if not isinstance(memory, (int, float)):
raise TypeError(
"memory in ray_actor_options must be an int or a float.")
elif memory < 0:
@ -118,8 +118,7 @@ class ReplicaConfig:
object_store_memory = self.ray_actor_options.get(
"object_store_memory", 0)
if not (isinstance(object_store_memory, int)
or isinstance(object_store_memory, float)):
if not isinstance(object_store_memory, (int, float)):
raise TypeError(
"object_store_memory in ray_actor_options must be "
"an int or a float.")