2020-03-11 09:09:56 +08:00

7 lines
184 B
Python

class Batch(object):
"""Suggested keys: [obs, act, rew, done, obs_next, info]"""
def __init__(self, **kwargs):
super().__init__()
self.__dict__.update(kwargs)