ray/rllib/agents/sac/__init__.py
ddworak94 fba8461663
[RLlib] Add RNN-SAC agent (#16577)
Shoutout to @ddworak94 :)
2021-07-25 10:04:52 -04:00

12 lines
534 B
Python

from ray.rllib.agents.sac.sac import SACTrainer, DEFAULT_CONFIG
from ray.rllib.agents.sac.sac_tf_policy import SACTFPolicy
from ray.rllib.agents.sac.sac_torch_policy import SACTorchPolicy
from ray.rllib.agents.sac.rnnsac import RNNSACTrainer, \
DEFAULT_CONFIG as RNNSAC_DEFAULT_CONFIG
from ray.rllib.agents.sac.rnnsac import RNNSACTorchPolicy
__all__ = [
"DEFAULT_CONFIG", "SACTFPolicy", "SACTorchPolicy", "SACTrainer",
"RNNSAC_DEFAULT_CONFIG", "RNNSACTorchPolicy", "RNNSACTrainer"
]