environment-dev.yml

# Conda environment for Performance RNN Development
# This includes all development tools (testing, formatting, documentation)
# Install with: conda env create -f environment-dev.yml
# Activate with: conda activate py_magenta
#
# Note: Choose your PyTorch installation based on your system:
# - For CPU: Use environment-cpu.yml instead, then install dev tools with: pip install -e ".[dev]"
# - For CUDA: Use environment-cuda.yml instead, then install dev tools with: pip install -e ".[dev]"
# - For MPS: Use environment-mps.yml instead, then install dev tools with: pip install -e ".[dev]"

name: py_magenta

channels:
  - pytorch
  - conda-forge
  - defaults

dependencies:
  - python=3.10

  # PyTorch (CPU-only for development - change based on your system)
  - pytorch>=2.0.0
  - torchvision
  - torchaudio
  - cpuonly

  # Core dependencies
  - numpy>=1.21.0,<2.0.0
  - pandas>=1.3.0
  - tqdm>=4.60.0
  - tensorboard>=2.10.0

  # Testing
  - pytest>=7.0.0
  - pytest-cov>=3.0.0
  - pytest-xdist>=2.5.0

  # Code formatting and linting
  - black>=22.0.0
  - isort>=5.10.0
  - flake8>=4.0.0
  - mypy>=0.950

  # Documentation
  - sphinx>=4.5.0
  - matplotlib>=3.5.0

  # Jupyter notebooks
  - jupyter>=1.0.0
  - ipykernel>=6.0.0

  # pip dependencies
  - pip
  - pip:
    - pretty_midi>=0.2.9
    - optuna>=3.0.0
    - flake8-docstrings>=1.6.0
    - flake8-bugbear>=22.0.0
    - types-tqdm
    - pre-commit>=2.17.0
    - sphinx-rtd-theme>=1.0.0
    - sphinx-autodoc-typehints>=1.18.0