[typing] update jaxlib & remove unnecessary ignore

MarMarhoun · Nov 4, 2022 · 0691be6 · 0691be6
1 parent 2f1f3e4
commit 0691be6
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
@@ -19,7 +19,7 @@ repos:
   - id: mypy
     files: (jax/|tests/typing_test\.py)
     exclude: jax/_src/basearray.py  # Use pyi instead
-    additional_dependencies: [types-requests==2.28.11, jaxlib==0.3.22]
+    additional_dependencies: [types-requests==2.28.11, jaxlib==0.3.24]
 
 - repo: https://github.com/mwouts/jupytext
   rev: v1.14.1

diff --git a/jax/interpreters/mlir.py b/jax/interpreters/mlir.py
@@ -1597,10 +1597,9 @@ def _wrapped_callback(token, *args):  # type: ignore  # pylint: disable=function
     ]
     operands = [token, *operands]
     result_types = [token_type()[0], *result_types]
-  # TODO(jakevdp) remove ignore statement once jaxlib>0.3.22 is released.
   callback_descriptor, keepalive = (
       backend.get_emit_python_callback_descriptor(_wrapped_callback,
-                                                  operand_shapes,  # type: ignore[arg-type]
+                                                  operand_shapes,
                                                   result_shapes))
   descriptor_operand = ir_constant(
       callback_descriptor, canonicalize_types=False)