ImportError: cannot import name 'oex_orchestration_pb2' from 'orbax.export.protos'

[kanglant]

When running the latest orbax-export, the from orbax.export.protos import oex_orchestration_pb2 line in utils.py is failing with an ImportError: cannot import name 'oex_orchestration_pb2' from 'orbax.export.protos'.

Here is a test log: https://github.com/jax-ml/jax-ai-stack/actions/runs/17402977130/job/49400221219.

[zg009]

I had this error and downgraded to orbax-export==0.0.6

[cpgaffney1]

Export-related tests are passing currently, in a recently-pushed PR. https://github.com/google/orbax/actions/runs/17745289200/job/50428988473?pr=2385

[jakevdp]

Thanks - will there be a new PyPI release with the fix?