agent/__init__.py

"""
Agent package.

Provides:
  BaseAgent        — abstract interface
  QLearningAgent   — tabular Q-learning (NumPy only)
  DQNAgent         — Deep Q-Network (PyTorch, GPU-accelerated)
"""

from .base_agent import BaseAgent
from .q_learning_agent import QLearningAgent

# DQN requires PyTorch
try:
    from .dqn_agent import DQNAgent
    DQN_AVAILABLE = True
except ImportError:
    DQN_AVAILABLE = False
    DQNAgent = None  # type: ignore

__all__ = ["BaseAgent", "QLearningAgent", "DQNAgent"]