Merge branch 'main' into 244-active-mask

Author: IFenton

.github/workflows/code_style.yaml

@@ -13,7 +13,7 @@ jobs:
     name: Format and lint code
     steps:
       - name: Checkout code
-        uses: actions/checkout@v6
+        uses: actions/checkout@v7
         with:
           fetch-depth: 0
       - name: Setup Python

.github/workflows/test_code.yaml

@@ -25,7 +25,7 @@ jobs:
         python-version: ["3.11", "3.12"]
     steps:
       - name: Checkout code
-        uses: actions/checkout@v6
+        uses: actions/checkout@v7
         with:
           fetch-depth: 0
       - name: Set up Python ${{ matrix.python-version }}

icenet_mp/__init__.py

@@ -1,7 +1,11 @@
+import logging
 import os
 import sys
 import warnings
 
+logging.getLogger("anemoi").setLevel(logging.WARNING)
+logging.getLogger("argopy").setLevel(logging.WARNING)
+
 os.environ["PYTORCH_ENABLE_MPS_FALLBACK"] = "1"
 if "torch" in sys.modules:
     warnings.warn(

icenet_mp/callbacks/ema_weight_averaging_callback.py

@@ -1,7 +1,6 @@
+from lightning.pytorch.callbacks import WeightAveraging
 from torch.optim.swa_utils import get_ema_multi_avg_fn
 
-from .weight_averaging import WeightAveraging
-
 
 class EMAWeightAveragingCallback(WeightAveraging):
     """A callback that updates an averaged model for Exponential Moving Average (EMA) after each training step."""