454c86c469
Things changed in this PR: - various docs update, add TOC - split buffer into several files - fix venv action_space randomness
29 lines
1002 B
Python
29 lines
1002 B
Python
from tianshou.data.batch import Batch
|
|
from tianshou.data.utils.converter import to_numpy, to_torch, to_torch_as
|
|
from tianshou.data.utils.segtree import SegmentTree
|
|
from tianshou.data.buffer.base import ReplayBuffer
|
|
from tianshou.data.buffer.prio import PrioritizedReplayBuffer
|
|
from tianshou.data.buffer.manager import ReplayBufferManager
|
|
from tianshou.data.buffer.manager import PrioritizedReplayBufferManager
|
|
from tianshou.data.buffer.vecbuf import VectorReplayBuffer
|
|
from tianshou.data.buffer.vecbuf import PrioritizedVectorReplayBuffer
|
|
from tianshou.data.buffer.cached import CachedReplayBuffer
|
|
from tianshou.data.collector import Collector, AsyncCollector
|
|
|
|
__all__ = [
|
|
"Batch",
|
|
"to_numpy",
|
|
"to_torch",
|
|
"to_torch_as",
|
|
"SegmentTree",
|
|
"ReplayBuffer",
|
|
"PrioritizedReplayBuffer",
|
|
"ReplayBufferManager",
|
|
"PrioritizedReplayBufferManager",
|
|
"VectorReplayBuffer",
|
|
"PrioritizedVectorReplayBuffer",
|
|
"CachedReplayBuffer",
|
|
"Collector",
|
|
"AsyncCollector",
|
|
]
|