Fixed import in plugin_executable.py

lukebaumann · copybara-github · commit 920985e734be · 2024-09-25T10:35:09.000-07:00
PiperOrigin-RevId: 678756888
diff --git a/pathwaysutils/plugin_executable.py b/pathwaysutils/plugin_executable.py
@@ -19,7 +19,7 @@
 
 import jax
 from jax._src.interpreters import pxla
-from jax._src.lib.xla_extension import ifrt_programs
+from jaxlib.xla_extension import ifrt_programs
 
 
 class PluginExecutable:
