Skip to content

Commit 920985e

Browse files
lukebaumanncopybara-github
authored andcommitted
Fixed import in plugin_executable.py
PiperOrigin-RevId: 678756888
1 parent 9fc2e96 commit 920985e

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

pathwaysutils/plugin_executable.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,7 @@
1919

2020
import jax
2121
from jax._src.interpreters import pxla
22-
from jax._src.lib.xla_extension import ifrt_programs
22+
from jaxlib.xla_extension import ifrt_programs
2323

2424

2525
class PluginExecutable:

0 commit comments

Comments
 (0)