fc251ab0b8
* add makefile * bump version * add isort and yapf * update contributing.md * update PR template * spelling check
36 lines
988 B
Python
36 lines
988 B
Python
"""Data package."""
|
|
# isort:skip_file
|
|
|
|
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,
|
|
PrioritizedReplayBufferManager,
|
|
)
|
|
from tianshou.data.buffer.vecbuf import (
|
|
VectorReplayBuffer,
|
|
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",
|
|
]
|