from ray.rllib.algorithms.dqn.dqn import DQN, DQNConfig, DEFAULT_CONFIG
from ray.rllib.algorithms.dqn.dqn_tf_policy import DQNTFPolicy
from ray.rllib.algorithms.dqn.dqn_torch_policy import DQNTorchPolicy

__all__ = [
    "DQN",
    "DQNConfig",
    "DQNTFPolicy",
    "DQNTorchPolicy",
    "DEFAULT_CONFIG",
]

from ray.rllib.algorithms.dqn.dqn import DQN, DQNConfig, DEFAULT_CONFIG
from ray.rllib.algorithms.dqn.dqn_tf_policy import DQNTFPolicy
from ray.rllib.algorithms.dqn.dqn_torch_policy import DQNTorchPolicy

__all__ = [
    "DQN",
    "DQNConfig",
    "DQNTFPolicy",
    "DQNTorchPolicy",
    "DEFAULT_CONFIG",
]
