Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
3 changes: 0 additions & 3 deletions .github/workflows/test_code.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -36,9 +36,6 @@ jobs:
uses: astral-sh/setup-uv@v6
with:
version: "0.8.3"
- name: Install ffmpeg (so matplotlib can call it)
run: |
sudo apt-get update && sudo apt-get install -y ffmpeg
- name: Run pytest
run: uv run --group dev pytest
- name: Run mypy
Expand Down
2 changes: 2 additions & 0 deletions icenet_mp/plugins.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,7 @@

from icenet_mp.data_processors.filters import register_filters
from icenet_mp.data_processors.sources import register_sources
from icenet_mp.visualisations import register_animation_backends
from icenet_mp.xpu import register_accelerators

logger = getLogger(__name__)
Expand All @@ -11,5 +12,6 @@ def register_plugins() -> None:
"""Register all plugins."""
logger.debug("Registering plugins with external libraries...")
register_accelerators()
register_animation_backends()
register_filters()
register_sources()
9 changes: 9 additions & 0 deletions icenet_mp/visualisations/__init__.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,15 @@
from imageio_ffmpeg import get_ffmpeg_exe
from matplotlib import rcParams

from .helpers import DEFAULT_SIC_SPEC
from .plotter import Plotter


def register_animation_backends() -> None:
"""Register the ImageIO FFMPEG animation backend."""
rcParams["animation.ffmpeg_path"] = get_ffmpeg_exe()


__all__ = [
"DEFAULT_SIC_SPEC",
"Plotter",
Expand Down
15 changes: 6 additions & 9 deletions pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -25,9 +25,10 @@ dependencies = [
"anemoi-utils>=0.4.42",
"cachetools>=6.1.0",
"cdsapi>=0.7.6",
"eccodeslib>=2.40.0", # indirect dependency of icenet that is not automatically installed
"eccodeslib>=2.40.0,<=2.45.3,!=2.42.0,!=2.43.0,!=2.44.0.2", # indirect dependency of icenet that is not automatically installed
"hydra-core>=1.3.2",
"icenet==0.2.7", # newer versions require netCDF4<1.6.1 which does not work on macOS 15
"imageio-ffmpeg>=0.6.0",
"jaxtyping>=0.3.2",
"lightning>=2.5.1",
"matplotlib>=3.10.3",
Expand Down Expand Up @@ -78,6 +79,7 @@ module = [
"anemoi.utils.*",
"earthkit.*",
"icenet.data.sic.*",
"imageio_ffmpeg.*",
"zarr.*",
]
ignore_missing_imports = true
Expand All @@ -91,16 +93,11 @@ addopts = [
testpaths = ["tests"]
norecursedirs = [
".git",
".venv",
"venv",
"__pycache__",
".pytest_cache",
".mypy_cache",
"node_modules",
".idea",
".pytest_cache",
".ruff_cache",
".venv",
".vscode",
"data",
"raw_input_*",
]

[tool.ruff]
Expand Down
Loading
Loading