Spaces:
Sleeping
Sleeping
File size: 1,239 Bytes
195f87e | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 | # Pin the versions called out in the plan (Section 1.1).
# These versions are known compatible with Python 3.11 + CUDA 12.x.
# DO NOT bump without re-running scripts/validate_env.py.
# --- Quantum simulation ---
stim>=1.13,<2.0
pymatching>=2.2,<3.0
# --- Environment / serving ---
fastapi>=0.110
uvicorn[standard]>=0.27
pydantic>=2.5,<3.0
httpx>=0.27
numpy>=1.26,<2.1
# --- Plotting (used by scripts/plot_results.py) ---
matplotlib>=3.8
pillow>=10
# --- Test runner ---
pytest>=8
# --- OpenEnv (HuggingFace's RL-env framework). Required by the submission
# guidelines ("Use OpenEnv (latest release). Build on top of the
# framework; don't reinvent the wheel."). Our server wraps the
# DecoderEnvironment with `openenv.core.Environment`-compatible
# Action/Observation/State models so TRL can drive it via
# `environment_factory=` (see qubit_medic/server/openenv_adapter.py
# and scripts/train_grpo.py). Lightweight; safe in the Spaces image.
openenv-core>=0.2.1
# --- Training (omit on CPU-only deploy) ---
# Heavy ML deps live in requirements-train.txt; the env server itself does
# not import any of them. Keeping them out of the Docker image keeps the
# Spaces upload under the free-tier image-size limit.
|