ray/rllib/agents/marwil/__init__.py
2022-05-21 12:50:20 +02:00

31 lines
776 B
Python

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 (
MARWILDynamicTFPolicy,
MARWILEagerTFPolicy,
)
from ray.rllib.algorithms.marwil.marwil_torch_policy import MARWILTorchPolicy
__all__ = [
"BCConfig",
"BCTrainer",
"MARWILConfig",
"MARWILDynamicTFPolicy",
"MARWILEagerTFPolicy",
"MARWILTorchPolicy",
"MARWILTrainer",
# Deprecated.
"BC_DEFAULT_CONFIG",
"DEFAULT_CONFIG",
]
from ray.rllib.utils.deprecation import deprecation_warning
deprecation_warning(
"ray.rllib.agents.marwil", "ray.rllib.algorithms.marwil", error=False
)