ray/rllib/algorithms/slateq/__init__.py

15 lines
369 B
Python

from ray.rllib.algorithms.slateq.slateq import (
SlateQ,
SlateQConfig,
DEFAULT_CONFIG,
)
from ray.rllib.algorithms.slateq.slateq_tf_policy import SlateQTFPolicy
from ray.rllib.algorithms.slateq.slateq_torch_policy import SlateQTorchPolicy
__all__ = [
"SlateQ",
"SlateQConfig",
"SlateQTFPolicy",
"SlateQTorchPolicy",
"DEFAULT_CONFIG",
]