batteryswapaichallenge/BatterySwapAI2026-Example

Setting up virtual environment

On Linux / Mac OS / Windows Subsystem for Linux (WSL)

python -m venv venv
source venv/bin/activate

Install dependencies

pip install -r requirements.txt -r requirements.dev.txt

Develop in virtual environment

Run the training

python batteryswap_example/train.py

Submitting trained models

Trained models or other output used by submission processing (script.py), must be committed in the git repository. An example is batteryswap_example/planners/best.pickle.

Test submissions in Docker - recommended

Using Docker allows to have exactly the same software versions as the submissions system.

This helps to ensure there are no errors when running in the submission environment.

NOTE: this requires around 20 GB+ of disk space.

Build Docker image

docker build -t batteryswapai-2026-example .

Make submissions and run evaluation

docker run --name batteryswapai -v ./dataset:/tmp/data batteryswapai-2026-example bash -c "/app/env/bin/python3 script.py && /app/env/bin/python3 -m batteryswap_public.metric"

Copy submission.csv out of container

docker cp batteryswapai:/app/submission.csv ./submission.csv