ray/rllib/algorithms/marwil/__init__.py

Ignoring revisions in .git-blame-ignore-revs. Click here to bypass and see the normal blame view.

25 lines
624 B
Python
Raw Normal View History

2022-05-21 03:50:20 -07:00
from ray.rllib.algorithms.marwil.bc import BCConfig, BCTrainer, BC_DEFAULT_CONFIG
from ray.rllib.algorithms.marwil.marwil import (
DEFAULT_CONFIG,
MARWILConfig,
MARWILTrainer,
)
from ray.rllib.algorithms.marwil.marwil_tf_policy import (
MARWILStaticGraphTFPolicy,
MARWILEagerTFPolicy,
)
from ray.rllib.algorithms.marwil.marwil_torch_policy import MARWILTorchPolicy
__all__ = [
2022-05-21 03:50:20 -07:00
"BCConfig",
"BCTrainer",
2022-05-21 03:50:20 -07:00
"MARWILConfig",
"MARWILStaticGraphTFPolicy",
"MARWILEagerTFPolicy",
"MARWILTorchPolicy",
"MARWILTrainer",
2022-05-21 03:50:20 -07:00
# Deprecated.
"BC_DEFAULT_CONFIG",
"DEFAULT_CONFIG",
]