| """Trainer package.""" | |
| from tianshou.trainer.base import ( | |
| BaseTrainer, | |
| OfflineTrainer, | |
| OffpolicyTrainer, | |
| OnpolicyTrainer, | |
| ) | |
| from tianshou.trainer.utils import gather_info, test_episode | |
| __all__ = [ | |
| "BaseTrainer", | |
| "OffpolicyTrainer", | |
| "OnpolicyTrainer", | |
| "OfflineTrainer", | |
| "test_episode", | |
| "gather_info", | |
| ] | |