Add queue size and __len__ methods (#2432)

This commit is contained in:
Peter Schafhalter
2018-07-19 17:04:42 -07:00
committed by Philipp Moritz
parent 4225ac5081
commit 400a3e5705
+8 -1
View File
@@ -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())