diff --git a/qwix/_src/providers/odml.py b/qwix/_src/providers/odml.py
index 3494ad1..9cd486d 100644
--- a/qwix/_src/providers/odml.py
+++ b/qwix/_src/providers/odml.py
@@ -495,7 +495,7 @@ def _get_attributes(
     }
     if zp is not None:
       # zero_point has to be int64 for ODML.
-      attributes['zero_point'] = np.asarray(zp, np.int64).flatten()  # pyrefly: ignore[bad-typed-dict-key]
+      attributes['zero_point'] = np.asarray(zp, np.int64).flatten()  # pyrefly: ignore[bad-assignment, bad-typed-dict-key]
     if quantization_dim is not None:
-      attributes['quantization_dimension'] = quantization_dim  # pyrefly: ignore[bad-typed-dict-key]
+      attributes['quantization_dimension'] = quantization_dim  # pyrefly: ignore[bad-assignment, bad-typed-dict-key]
     return attributes