diff --git a/training/src/anemoi/training/losses/combined.py b/training/src/anemoi/training/losses/combined.py index b814a6a7..70b72f9b 100644 --- a/training/src/anemoi/training/losses/combined.py +++ b/training/src/anemoi/training/losses/combined.py @@ -17,8 +17,8 @@ from omegaconf import DictConfig from anemoi.training.losses.base import BaseLoss -from anemoi.training.losses.utils import ScaleTensor from anemoi.training.losses.loss import get_loss_function +from anemoi.training.losses.utils import ScaleTensor if TYPE_CHECKING: import torch