# Generated from environment.yml for pip-based installs.
# Usage: pip install -r requirements.txt
#
# This file covers pip-installable packages only. System dependencies such as
# swig are not handled here; see Dockerfile for the containerised setup.
#
# CPU PyTorch is the default. For GPU support, install the matching CUDA wheels
# first, then install the rest of this file:
#   pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121

--extra-index-url https://download.pytorch.org/whl/cpu

# Python and notebook tooling
notebook
ipywidgets

# Core deep learning
torch>=2.3
torchvision
torchaudio
pytorch-ignite
torchrl
ptan

# General scientific stack
numpy
matplotlib
pandas
scipy

# Image and video
opencv-python
pillow
moviepy

# RL environments and frameworks
gymnasium[mujoco,box2d]>=1.0.0
minigrid
ale-py
pygame
pettingzoo
stable-baselines3
sb3-contrib
ray[rllib,tune]>=2.54.0
mlflow
supersuit
shimmy
tensorboard

# Testing and type checking
pytest
mypy