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))
11 lines
276 B
Python
11 lines
276 B
Python
from tianshou.trainer.utils import test_episode, gather_info
|
|
from tianshou.trainer.onpolicy import onpolicy_trainer
|
|
from tianshou.trainer.offpolicy import offpolicy_trainer
|
|
|
|
__all__ = [
|
|
"gather_info",
|
|
"test_episode",
|
|
"onpolicy_trainer",
|
|
"offpolicy_trainer",
|
|
]
|