fix upstream

Author: gboehl

econpizza/parser/__init__.py

@@ -136,7 +136,7 @@ def _load_external_functions_file(model, context):
         module = _load_as_module(model["functions_file"])
 
         def func_or_compiled(func): return isinstance(
-            func, jaxlib.xla_extension.PjitFunction) or isfunction(func)
+            func, jaxlib._jax.PjitFunction) or isfunction(func)
         for m in getmembers(module, func_or_compiled):
             context[m[0]] = m[1]