ray/rllib/agents/slateq/__init__.py

8 lines
230 B
Python

from ray.rllib.agents.slateq.slateq import SlateQTrainer, DEFAULT_CONFIG
from ray.rllib.agents.slateq.slateq_torch_policy import SlateQTorchPolicy
__all__ = [
"SlateQTrainer",
"SlateQTorchPolicy",
"DEFAULT_CONFIG",
]