2022-05-16 00:45:32 -07:00
|
|
|
from ray.rllib.algorithms.marwil.bc import BCTrainer, BC_DEFAULT_CONFIG
|
|
|
|
from ray.rllib.algorithms.marwil.marwil import MARWILTrainer, DEFAULT_CONFIG
|
2022-05-20 05:10:59 -07:00
|
|
|
from ray.rllib.algorithms.marwil.marwil_tf_policy import (
|
|
|
|
MARWILDynamicTFPolicy,
|
|
|
|
MARWILEagerTFPolicy,
|
|
|
|
)
|
2022-05-16 00:45:32 -07:00
|
|
|
from ray.rllib.algorithms.marwil.marwil_torch_policy import MARWILTorchPolicy
|
|
|
|
|
|
|
|
__all__ = [
|
|
|
|
"BCTrainer",
|
|
|
|
"BC_DEFAULT_CONFIG",
|
|
|
|
"DEFAULT_CONFIG",
|
2022-05-20 05:10:59 -07:00
|
|
|
"MARWILDynamicTFPolicy",
|
|
|
|
"MARWILEagerTFPolicy",
|
2022-05-16 00:45:32 -07:00
|
|
|
"MARWILTorchPolicy",
|
|
|
|
"MARWILTrainer",
|
|
|
|
]
|