diff --git a/src/anemoi/graphs/edges/attributes.py b/src/anemoi/graphs/edges/attributes.py index 73079e0..b76d6d1 100644 --- a/src/anemoi/graphs/edges/attributes.py +++ b/src/anemoi/graphs/edges/attributes.py @@ -158,14 +158,14 @@ def post_process(self, values: np.ndarray) -> torch.Tensor: return values -class BooleanBaseEdgeAttribute(ABC, BaseEdgeAttribute): +class BooleanBaseEdgeAttribute(BaseEdgeAttribute, ABC): """Base class for boolean edge attributes.""" def __init__(self) -> None: super().__init__(norm=None, dtype="bool") -class AttributeFromNode(ABC, BooleanBaseEdgeAttribute): +class AttributeFromNode(BooleanBaseEdgeAttribute, ABC): """ Base class for Attribute from Node.