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

icedoom888 · JPXKQX · web-flow · commit a58dae6b864e · 2024-12-18T18:51:12.000+01:00
Co-authored-by: Mario Santa Cruz &lt;48736305+JPXKQX@users.noreply.github.com&gt;
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:
