⚠️ WARNING: Active Development
PatchSim is currently under heavy development. Features are experimental and subject to change. Not recommended for production use at this stage.
PatchSim is a modular metapopulation simulation framework for multi-disease epidemiological modelling.
To develop a general-purpose, modular simulation framework for patch-based metapopulation epidemiology, enabling modellers and researchers to simulate disease transmission under diverse scenarios, diseases, and intervention strategies. The framework balances robust scientific modelling with flexibility for exploratory research and translational use cases.
PatchSim combines metapopulation network dynamics with a lightweight, configuration-first workflow:
- Network-aware compartment transitions across connected patches
- Arrow-map transition syntax in YAML for explicit model specification
- Fast iteration loop from config edits to reproducible outputs
Compared to many epidemiology tools that are either code-heavy or tightly bound to a specific disease model, PatchSim offers:
- Model flexibility: define SIR/SIRS-style variants through config transitions
- Research-friendly reproducibility: deterministic inputs, logged runs, and versioned config files
- Dual interface: SDK import for programmatic workflows and CLI for operational runs
- Extensibility path: built-in structure for adding custom models and project templates
PatchSim aims to support a range of modelling features commonly used in metapopulation disease simulations:
- 🗺️ Spatial Networks: Represent geographical units (e.g., subdistricts, regions) as interconnected patches with movement/contact matrices.
- 👥 Stratification by Population Attributes:
- Age groups
- Species (e.g., cattle, buffalo)
- Risk groups or occupations
- 🧪 Disease Agnostic Compartment Models:
- SIR, SEIR, SIRS and extensions
- Supports both discrete timestep and ODE-based solvers
- 🛠️ Scenario and Parameter Management:
- Batch simulations for scenario comparison
- Sensitivity analysis and parameter sweeps
- 🧵 Reproducibility:
- Random seed control and metadata logging
- Version-tracked configurations
- 📦 Modularity:
- Plug-in architecture for solvers, interventions, and input data pipelines
Install from PyPI:
pip install patchsimInstall from source using uv:
# Clone the repository
git clone https://github.com/dsih-artpark/patchsim
cd patchsim
# Create a virtual environment and install dependencies
uv venv
source .venv/bin/activate
uv pip install -e .
# For development (with dev dependencies)
uv pip install -e .[dev]PatchSim provides a subcommand-based CLI. Always run using uv run to ensure correct dependency resolution:
# Show help and available options
uv run patchsim --help
# Show package version
uv run patchsim --version
# Initialize a new self-contained project
uv run patchsim init my-project
# Validate config without running
uv run patchsim validate -c my-project/config.yaml
# Run simulation
uv run patchsim run -c my-project/config.yaml
# List built-in models
uv run patchsim list-modelsimport patchsim
config = patchsim.load_config("config.yaml")
net, y0, patches, n = patchsim.setup_simulation(config)
patchsim.run_simulation(config, "my-model", net, y0, patches, n)The simulation outputs are saved in the following structure:
output/
├── logs/
│ └── cli_YYYYMMDD_HHMMSS.log # Timestamped log files
└── sample-sir-ode/ # Model-specific output directory
├── plots/
│ └── patch_timeseries_sample-sir-ode_ode.png
└── runs/
└── all_patches_sample-sir-ode_ode.csv
Simulations are configured using YAML files. The configuration file specifies:
- Model parameters (e.g., transmission rates)
- Input files (patch populations, seed data, network)
- Simulation settings (time horizon, output directory)
Example configuration:
# Model parameters
Beta: 0.3
Gamma: 0.1
# Input files
PatchFile: data/patch/sample-sir-ode-patch-population.csv
SeedFile: data/seeds/sample-sir-ode-patchA-2.csv
NetworkFile: data/networks/sample-network-static.csv
# Simulation settings
OutputDir: output/sample-sir-ode
TMax: 50We welcome contributions!
To contribute: fork the repo, create a branch, make your changes, and open a pull request.
For major changes, please open an issue first to discuss.
Thanks for helping improve the framework!
This project is licensed under the GNU General Public License v3.0.
You may use, modify, and share this project under the same license terms. See the LICENSE file for full details.
