Tianshou/tianshou/data/__init__.py

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 import ReplayBuffer, \
    ListReplayBuffer, PrioritizedReplayBuffer
from tianshou.data.collector import Collector

__all__ = [
    "Batch",
    "to_numpy",
    "to_torch",
    "to_torch_as",
    "SegmentTree",
    "ReplayBuffer",
    "ListReplayBuffer",
    "PrioritizedReplayBuffer",
    "Collector",
]
env and data 2020-03-11 09:09:56 +08:00			`from tianshou.data.batch import Batch`
code format and update function signatures (#213) Cherry-pick from #200 - update the function signature - format code-style - move _compile into separate functions - fix a bug in to_torch and to_numpy (Batch) - remove None in action_range In short, the code-format only contains function-signature style and `'` -> `"`. (pick up from [black](https://github.com/psf/black)) 2020-09-12 15:39:01 +08:00			`from tianshou.data.utils.converter import to_numpy, to_torch, to_torch_as`
Improve PER (#159) - use segment tree to rewrite the previous PrioReplayBuffer code, add the test - enable all Q-learning algorithms to use PER 2020-08-06 10:26:24 +08:00			`from tianshou.data.utils.segtree import SegmentTree`
add ListReplayBuffer 2020-03-28 15:14:41 +08:00			`from tianshou.data.buffer import ReplayBuffer, \`
			`ListReplayBuffer, PrioritizedReplayBuffer`
half of collector 2020-03-12 22:20:33 +08:00			`from tianshou.data.collector import Collector`
env and data 2020-03-11 09:09:56 +08:00
maybe finished collector? 2020-03-13 17:49:22 +08:00			`__all__ = [`
code format and update function signatures (#213) Cherry-pick from #200 - update the function signature - format code-style - move _compile into separate functions - fix a bug in to_torch and to_numpy (Batch) - remove None in action_range In short, the code-format only contains function-signature style and `'` -> `"`. (pick up from [black](https://github.com/psf/black)) 2020-09-12 15:39:01 +08:00			`"Batch",`
			`"to_numpy",`
			`"to_torch",`
			`"to_torch_as",`
			`"SegmentTree",`
			`"ReplayBuffer",`
			`"ListReplayBuffer",`
			`"PrioritizedReplayBuffer",`
			`"Collector",`
maybe finished collector? 2020-03-13 17:49:22 +08:00			`]`