8af7196a9a
* Enable to convert Batch data back to torch. * Add torch converter to collector. * Fix * Move to_numpy/to_torch convert in dedicated utils.py. * Use to_numpy/to_torch to convert arrays. * fix lint * fix * Add unit test to check Batch from/to numpy. * Fix Batch over Batch. Co-authored-by: Alexis Duburcq <alexis.duburcq@wandercraft.eu>
16 lines
381 B
Python
16 lines
381 B
Python
from tianshou.data.batch import Batch
|
|
from tianshou.data.utils import to_numpy, to_torch
|
|
from tianshou.data.buffer import ReplayBuffer, \
|
|
ListReplayBuffer, PrioritizedReplayBuffer
|
|
from tianshou.data.collector import Collector
|
|
|
|
__all__ = [
|
|
'Batch',
|
|
'to_numpy',
|
|
'to_torch',
|
|
'ReplayBuffer',
|
|
'ListReplayBuffer',
|
|
'PrioritizedReplayBuffer',
|
|
'Collector'
|
|
]
|