Bris inference

This is a package to run MET Norway's data-driven model Bris, which is based on the Anemoi framework.

Features

• Model and data-parallel inference
• Multi encoder/decoder
• Time interpolation

Documentation

See Wiki

Requirements

• udunits2 library. On ubuntu available as libudunits2-0

Install

Locally for development

python3 -m venv venv && source venv/bin/activate
pip install -e .

Via docker, if you are from Met.no

See Dockerfile

How to run tests

pip install ".[tests]"
tox

When pushing to github, these will be run automatically and must succeed. Read more about Tests in the wiki.

Code borrowed from Anemoi project

• bris/ddp_strategy.py is based on https://github.com/ecmwf/anemoi-core/blob/main/training/src/anemoi/training/distributed/strategy.py
• bris/grid_indices.py is based on https://github.com/ecmwf/anemoi-core/blob/main/training/src/anemoi/training/data/grid_indices.py
• bris/data/data{set,module}.py is somewhat based on https://github.com/ecmwf/anemoi-core/tree/main/training/src/anemoi/training/data