mirror of
https://github.com/wassname/ray.git
synced 2026-07-10 18:29:20 +08:00
c2cb5c2214
* WIP. * WIP. * LINT. * Fix MARWIL so it can run with eager-mode. * LINT.
11 lines
320 B
Python
11 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",
|
|
]
|