From 400a3e5705f4c98acf528df388d175262af903c4 Mon Sep 17 00:00:00 2001 From: Peter Schafhalter Date: Thu, 19 Jul 2018 17:04:42 -0700 Subject: [PATCH] Add queue size and __len__ methods (#2432) --- python/ray/experimental/queue.py | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/python/ray/experimental/queue.py b/python/ray/experimental/queue.py index 39961db08..d2cc9f434 100644 --- a/python/ray/experimental/queue.py +++ b/python/ray/experimental/queue.py @@ -27,10 +27,17 @@ class Queue(object): self.maxsize = maxsize self.actor = _QueueActor.remote(maxsize) - def qsize(self): + def __len__(self): + return self.size() + + def size(self): """The size of the queue.""" return ray.get(self.actor.qsize.remote()) + def qsize(self): + """The size of the queue.""" + return self.size() + def empty(self): """Whether the queue is empty.""" return ray.get(self.actor.qsize.remote())