Metadata-Version: 2.4
Name: drl-wizard
Version: 0.1.3
Summary: Deep Reinforcement Learning training platform
Author: Amin Lotfolahi
License: MIT
Project-URL: Homepage, https://github.com/amin-lotf/DRL-Wizard
Requires-Python: >=3.9
Description-Content-Type: text/markdown
Requires-Dist: sqlalchemy>=2.0
Requires-Dist: aiosqlite>=0.20
Requires-Dist: tensorboard>=2.14
Requires-Dist: gymnasium[atari,mujoco]>=0.29
Requires-Dist: gymnasium-robotics>=1.2
Requires-Dist: mpe2>=0.0.1
Requires-Dist: fastapi[standard]>=0.115
Requires-Dist: uvicorn[standard]>=0.30
Requires-Dist: pandas>=2.2
Requires-Dist: stable-baselines3>=2.3
Requires-Dist: ale-py>=0.10
Requires-Dist: opencv-python>=4.9
Provides-Extra: ui
Requires-Dist: streamlit>=1.38; extra == "ui"
Requires-Dist: pandas-stubs>=2.2; extra == "ui"
Provides-Extra: dev
Requires-Dist: pytest>=8; extra == "dev"
Requires-Dist: pytest-asyncio>=0.23; extra == "dev"
Requires-Dist: ruff>=0.6; extra == "dev"
Requires-Dist: mypy>=1.10; extra == "dev"

# DRL Wizard

**DRL Wizard** is a clean, modular Deep Reinforcement Learning toolkit for training, comparing, and understanding modern RL algorithms. It supports multiple environments, multiple algorithms, real-time monitoring, and a unified workflow through a FastAPI backend and Streamlit UI.

---

## 🚀 Features

- **Algorithms:** PPO, TRPO, DQN, Double DQN, Dueling DQN, SAC (more coming)
- **Environments:** Gymnasium, Atari (ALE), image-based & multi-discrete action spaces
- **Architecture:** FastAPI backend, Streamlit UI, SQLAlchemy storage, Pydantic configs
- **Experiment Tools:** concurrent jobs, graceful stop, NDJSON logs, TensorBoard, job archives
- **Extensible:** easy to add algorithms, envs, or visualization components

---

---

## 📦 Installation
Basic
```bash
pip install drl-wizard
```

UI + Dev tools:
```bash
pip install drl-wizard[ui,dev]
```

## 🖥️ Running
 - **Running the UI & Backend:**
```bash
drl-wizard-run
```

 - **Running Backend:**
```bash
drl-wizard-api
```

 - **Running UI:**
```bash
drl-wizard-ui
```

## 📝 Notes
- drl-wizard-run launches the full platform.

- Extras ([ui], [dev]) include Streamlit, and development dependencies.
