From 3e77709f6426e17a520b2fb6e664c26386d4f1fc Mon Sep 17 00:00:00 2001 From: Jake VanderPlas Date: Tue, 3 Jun 2025 14:09:15 -0700 Subject: [PATCH] Avoid private import of `jax._src.interpreters` PiperOrigin-RevId: 766814939 --- pathwaysutils/plugin_executable.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pathwaysutils/plugin_executable.py b/pathwaysutils/plugin_executable.py index 21f1b44..7d28536 100644 --- a/pathwaysutils/plugin_executable.py +++ b/pathwaysutils/plugin_executable.py @@ -18,7 +18,7 @@ from typing import List, Sequence, Tuple, Union import jax -from jax._src.interpreters import pxla +from jax.interpreters import pxla from jax.extend.ifrt_programs import ifrt_programs