Import failure 'No module named 'substrait.gen.proto'

[constantingoeldel]

What happened?

When running

uv run --with vortex-data python3
Python 3.13.5 (main, Jun 12 2025, 12:22:43) [Clang 20.1.4 ] on darwin
Type "help", "copyright", "credits" or "license" for more information.
>>> import vortex as vx

I get the following import error:

Traceback (most recent call last):
  File "<python-input-0>", line 1, in <module>
    import vortex as vx
  File "/Users/<user>/.cache/uv/archive-v0/XAj_xg23woHlhB7nKFWBb/lib/python3.13/site-packages/vortex/__init__.py", line 6, in <module>
    from . import _lib, arrays, dataset, expr, file, io, ray, registry, scan
  File "/Users/<user>/.cache/uv/archive-v0/XAj_xg23woHlhB7nKFWBb/lib/python3.13/site-packages/vortex/dataset.py", line 18, in <module>
    from .arrow.expression import ensure_vortex_expression
  File "/Users/<user>/.cache/uv/archive-v0/XAj_xg23woHlhB7nKFWBb/lib/python3.13/site-packages/vortex/arrow/expression.py", line 14, in <module>
    from ..substrait import extended_expression
  File "/Users/<user>/.cache/uv/archive-v0/XAj_xg23woHlhB7nKFWBb/lib/python3.13/site-packages/vortex/substrait.py", line 8, in <module>
    from substrait.gen.proto.algebra_pb2 import Expression, FunctionArgument
ModuleNotFoundError: No module named 'substrait.gen.proto'

Steps to reproduce

uv run --with vortex-data python3
import vortex as vx

Environment

• Vortex Version 0.58.0
• Python 3.13.5
• Macos 15.7.3

Additional context

No response

[blaginin]

This should unblock this: #6216

While there's no Vortex release, you need to manually constrain substrait version, e.g. by having this as project dependencies:

dependencies = [
"vortex-data",
"substrait<0.27"
]