diff --git a/src/metrax/classification_metrics.py b/src/metrax/classification_metrics.py
index 31d4302..304ac2b 100644
--- a/src/metrax/classification_metrics.py
+++ b/src/metrax/classification_metrics.py
@@ -711,7 +711,7 @@ class BinaryAccuracy(base.Average):
   """
 
   @classmethod
-  def from_model_output(
+  def from_model_output(  # pyrefly: ignore[bad-override]
       cls,
       predictions: jax.Array,
       labels: jax.Array,
@@ -752,7 +752,7 @@ class CategoricalAccuracy(base.Average):
   """
 
   @classmethod
-  def from_model_output(
+  def from_model_output(  # pyrefly: ignore[bad-override]
       cls,
       predictions: jax.Array,
       labels: jax.Array,
@@ -790,7 +790,7 @@ class SparseCategoricalAccuracy(base.Average):
   """
 
   @classmethod
-  def from_model_output(
+  def from_model_output(  # pyrefly: ignore[bad-override]
       cls,
       predictions: jax.Array,
       labels: jax.Array,