ray/rllib/algorithms/marwil/__init__.py

24 lines
624 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",
]