Merge pull request #18943 from jakevdp:xla-bridge

PiperOrigin-RevId: 590269567
google · Dec 12, 2023 · 2fa90e1 · 2fa90e1
2 parents 616f4d2 + 796449e
commit 2fa90e1
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/jax/_src/xla_bridge.py b/jax/_src/xla_bridge.py
@@ -18,6 +18,7 @@
 and provide some automatic type mapping logic for converting between Numpy and
 XLA. There are also a handful of related casting utilities.
 """
+from __future__ import annotations
 
 from collections.abc import Mapping
 import dataclasses