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
)