Update src/anemoi/graphs/edges/attributes.py

Co-authored-by: Mario Santa Cruz <48736305+JPXKQX@users.noreply.github.com>
MeteoSwiss · Dec 18, 2024 · a58dae6 · a58dae6
1 parent 77b2335
commit a58dae6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/anemoi/graphs/edges/attributes.py b/src/anemoi/graphs/edges/attributes.py
@@ -158,7 +158,7 @@ def post_process(self, values: np.ndarray) -> torch.Tensor:
         return values
 
 
-class BooleanBaseEdgeAttribute(BaseEdgeAttribute):
+class BooleanBaseEdgeAttribute(ABC, BaseEdgeAttribute):
     """Base class for boolean edge attributes."""
 
     def __init__(self) -> None: