Spaces:
Sleeping
Sleeping
| """ | |
| 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"] | |