Example Model for BatterySwapAI 2026
This repository shows how to create a solution for the BatterySwapAI 2026 challenge.
It contains working example code that can be submitted as-is, along with tools to check the solution before submitting.
Setup
Prerequisites
You need to have the following installed
- git
- Python 3.10+
- Docker
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
Developing
Develop in virtual environment
Run the training
python batteryswap_example/train.py
Making submissions
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
Create new submission
NOTE: remember to commit and push your changes to the HuggingFace model repository.
Use New submission in the competition application to submit your current code for evaluation.