State Transition model checkpoints are meant to be used with the Arc State package.
State is distributed via uv. To train models on your own AnnDatas, please see the repository README.
Running inference from pypi
uv tool install arc-state
To
state tx infer --model_dir <ST-Parse_PATH> --pert_col cytokine --adata <INPUT_ADATA>.h5ad --output <OUTPUT_DIR>
Running from source
# Clone repo
git clone github.com:arcinstitute/state
cd state
# Initialize venv
uv venv
# Install
uv tool install -e .
To generate embeddings given an AnnData:
state tx infer --model_dir <ST-Parse_PATH> --pert_col cytokine --adata <INPUT_ADATA>.h5ad --output <OUTPUT_DIR>
For model licenses please see MODEL_ACCEPTABLE_USE_POLICY.md, MODEL_LICENSE.md, and LICENSE.md.