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.