"""PyTorch Q-functions."""
from garage.torch.q_functions.continuous_mlp_q_function import (
    ContinuousMLPQFunction)
from garage.torch.q_functions.discrete_cnn_q_function import (
    DiscreteCNNQFunction)
from garage.torch.q_functions.discrete_dueling_cnn_q_function import (
    DiscreteDuelingCNNQFunction)
from garage.torch.q_functions.discrete_mlp_q_function import (
    DiscreteMLPQFunction)
from garage.torch.q_functions.categorical_dueling_cnn_q_function import (
    CategoricalDuelingCNNQFunction
)

__all__ = [
    'ContinuousMLPQFunction', 'DiscreteCNNQFunction',
    'DiscreteDuelingCNNQFunction', 'DiscreteMLPQFunction',
    'CategoricalDuelingCNNQFunction'
]
