ray/rllib/agents/marwil/__init__.py
Sven Mika c2cb5c2214
[RLlib] MARWIL torch. (#7836)
* WIP.

* WIP.

* LINT.

* Fix MARWIL so it can run with eager-mode.

* LINT.
2020-04-06 16:38:50 -07:00

10 lines
320 B
Python

from ray.rllib.agents.marwil.marwil import MARWILTrainer, DEFAULT_CONFIG
from ray.rllib.agents.marwil.marwil_tf_policy import MARWILTFPolicy
from ray.rllib.agents.marwil.marwil_torch_policy import MARWILTorchPolicy
__all__ = [
"DEFAULT_CONFIG",
"MARWILTFPolicy",
"MARWILTorchPolicy",
"MARWILTrainer",
]