jaxlib: Add ifrt_proxy.pyi to build_wheel.py.

PiperOrigin-RevId: 617275734
jax-ml · Mar 19, 2024 · df9cefa · df9cefa
1 parent 6fa75aa
commit df9cefa
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/jaxlib/tools/build_wheel.py b/jaxlib/tools/build_wheel.py
@@ -100,6 +100,7 @@ def patch_copy_mlir_import(src_file, dst_dir):
 
 _XLA_EXTENSION_STUBS = [
     "__init__.pyi",
+    "ifrt_proxy.pyi",
     "jax_jit.pyi",
     "ops.pyi",
     "outfeed_receiver.pyi",