mirror of
https://github.com/vale981/ray
synced 2025-03-08 19:41:38 -05:00
12 lines
356 B
Python
12 lines
356 B
Python
|
from collections import namedtuple
|
||
|
|
|
||
|
|
|
||
|
|
class TupleActions(namedtuple("TupleActions", ["batches"])):
|
||
|
|
"""Used to return tuple actions as a list of batches per tuple element."""
|
||
|
|
|
||
|
|
def __new__(cls, batches):
|
||
|
|
return super(TupleActions, cls).__new__(cls, batches)
|
||
|
|
|
||
|
|
def numpy(self):
|
||
|
|
return TupleActions([b.numpy() for b in self.batches])
|