monsoon-rl / requirements.txt
DHDRL's picture
Update requirements.txt
b604b08 verified
Raw
History Blame
804 Bytes
# Core (needed for the data/scoring/Indonesia stack -- climatology.py,
# indonesia_zones.py, backtest_indonesia.py, era5_data_pipeline.py,
# zone_observation.py, crop_risk_scorer.py, hierarchical_search.py)
numpy>=1.24
# RL training stack (train_curriculum.py, train_kaggle.py,
# weather_forecast_env.py, gru_weather_policy.py)
torch>=2.0
gymnasium>=0.29
stable-baselines3>=2.0
sb3-contrib>=2.0
# Hyperparameter sweeps (sweep_reward_shaping.py)
optuna>=3.5
# Edge export (mnn_export.py) -- MNN itself has no pip package; build/install
# per https://github.com/alibaba/MNN, this only covers the ONNX/export side.
onnx>=1.15
# Optional -- only needed if you actually connect to a broker
# (node_transport.py's MQTTTransport falls back to LocalTransport without it)
paho-mqtt>=1.6
# Testing
pytest>=7.0