fix: avoid ruff linter error EM101

ecmwf · Oct 18, 2024 · 2fa1d5c · 2fa1d5c
1 parent 331317c
commit 2fa1d5c
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/anemoi/training/utils/masks.py b/src/anemoi/training/utils/masks.py
@@ -23,11 +23,13 @@ class BaseMask:
 
     @abstractmethod
     def apply(self, x: torch.Tensor, *args, **kwargs) -> torch.Tensor:
-        raise NotImplementedError("Method `apply` must be implemented in subclass.")
+        error_message = "Method `apply` must be implemented in subclass."
+        raise NotImplementedError(error_message)
 
     @abstractmethod
     def rollout_boundary(self, x: torch.Tensor, *args, **kwargs) -> torch.Tensor:
-        raise NotImplementedError("Method `rollout_boundary` must be implemented in subclass.")
+        error_message = "Method `rollout_boundary` must be implemented in subclass."
+        raise NotImplementedError(error_message)
 
 
 class Boolean1DMask(BaseMask):