ray/rllib/agents/marwil/__init__.py

31 lines
784 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 (
MARWILStaticGraphTFPolicy,
MARWILEagerTFPolicy,
)
from ray.rllib.algorithms.marwil.marwil_torch_policy import MARWILTorchPolicy
__all__ = [
"BCConfig",
"BCTrainer",
"MARWILConfig",
"MARWILStaticGraphTFPolicy",
"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
)