diff --git a/src/anemoi/models/models/encoder_processor_decoder.py b/src/anemoi/models/models/encoder_processor_decoder.py index b9995c7..41adb80 100644 --- a/src/anemoi/models/models/encoder_processor_decoder.py +++ b/src/anemoi/models/models/encoder_processor_decoder.py @@ -23,6 +23,7 @@ from anemoi.models.distributed.shapes import get_shape_shards from anemoi.models.layers.graph import TrainableTensor from anemoi.models.models.bounding import BaseBoundingStrategy + LOGGER = logging.getLogger(__name__)