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,
|
|
|
|
)
|
2022-05-20 05:10:59 -07:00
|
|
|
from ray.rllib.algorithms.marwil.marwil_tf_policy import (
|
2022-05-25 05:38:03 -07:00
|
|
|
MARWILStaticGraphTFPolicy,
|
2022-05-20 05:10:59 -07:00
|
|
|
MARWILEagerTFPolicy,
|
|
|
|
)
|
2022-05-16 00:45:32 -07:00
|
|
|
from ray.rllib.algorithms.marwil.marwil_torch_policy import MARWILTorchPolicy
|
|
|
|
|
|
|
|
__all__ = [
|
2022-05-21 03:50:20 -07:00
|
|
|
"BCConfig",
|
2022-05-16 00:45:32 -07:00
|
|
|
"BCTrainer",
|
2022-05-21 03:50:20 -07:00
|
|
|
"MARWILConfig",
|
2022-05-25 05:38:03 -07:00
|
|
|
"MARWILStaticGraphTFPolicy",
|
2022-05-20 05:10:59 -07:00
|
|
|
"MARWILEagerTFPolicy",
|
2022-05-16 00:45:32 -07:00
|
|
|
"MARWILTorchPolicy",
|
|
|
|
"MARWILTrainer",
|
2022-05-21 03:50:20 -07:00
|
|
|
# Deprecated.
|
|
|
|
"BC_DEFAULT_CONFIG",
|
|
|
|
"DEFAULT_CONFIG",
|
2022-05-16 00:45:32 -07:00
|
|
|
]
|