pyspiral 0.6.3__cp310-abi3-macosx_11_0_arm64.whl → 0.6.5__cp310-abi3-macosx_11_0_arm64.whl

spiral/core/authn/__init__.pyi

@@ -8,7 +8,7 @@ class Authn:
     @staticmethod
     def from_token(token: Token) -> Authn: ...
     @staticmethod
-    def from_fallback() -> Authn: ...
+    def from_fallback(api_url: str) -> Authn: ...
     @staticmethod
     def from_device() -> Authn: ...
     def token(self) -> Token | None: ...

spiral/core/client/__init__.pyi

@@ -26,12 +26,12 @@ class Spiral:
         projection: Expr,
         filter: Expr | None = None,
         asof: int | None = None,
-        exclude_keys: bool = False,
+        exclude_keys: bool | None = None,
     ) -> Scan:
         """Construct a table scan."""
         ...
 
-    def transaction(self, table: Table, format: str | None = None) -> Transaction:
+    def transaction(self, table: Table, format: str | None = None, retries: int | None = 3) -> Transaction:
         """Being a table transaction."""
         ...
 
@@ -136,11 +136,18 @@ class ShuffleStrategy:
     # Externally provided shards to shuffle before reading rows.
     shards: list[Shard] | None
 
+    # Maximum number of rows to return in a single batch.
+    # If None, it is derived from the shuffle buffer size.
+    # IMPORTANT: The returned batch may be smaller than this size.
+    max_batch_size: int | None
+
     def __init__(
         self,
+        shuffle_buffer_size: int,
+        *,
         seed: int | None = None,
-        shuffle_buffer_size: int | None = None,
         shards: list[Shard] | None = None,
+        max_batch_size: int | None = None,
     ): ...
 
 class Operations:
@@ -205,3 +212,7 @@ class Operations:
         """
         ...
     def metrics(self) -> dict[str, Any]: ...
+
+def flush_telemetry() -> None:
+    """Flush telemetry data to the configured exporter."""
+    ...

spiral/core/table/__init__.pyi

@@ -70,6 +70,9 @@ class Scan:
         self,
         strategy: ShuffleStrategy | None = None,
         batch_readahead: int | None = None,
+        num_workers: int | None = None,
+        worker_id: int | None = None,
+        infinite: bool = False,
     ) -> pa.RecordBatchReader: ...
     def metrics(self) -> dict[str, Any]: ...
     def _prepare_shard(

spiral/debug/manifests.py

@@ -1,3 +1,6 @@
+from rich.console import Console
+from rich.table import Table
+
 from spiral import datetime_
 from spiral.core.table import Scan
 from spiral.core.table.manifests import FragmentManifest
@@ -42,23 +45,26 @@ def _table_of_fragments(manifest: FragmentManifest, title: str):
     avg_size = total_size / fragment_count if fragment_count > 0 else 0
 
     # Print title and summary
-    print(f"\n\n{title}")
-    print(
+    console = Console()
+    console.print(f"\n\n{title}")
+    console.print(
         f"{fragment_count} fragments, "
         f"total: {_format_bytes(total_size)}, "
         f"avg: {_format_bytes(int(avg_size))}, "
         f"metadata: {_format_bytes(total_metadata_size)}"
     )
-    print("=" * 120)
 
-    # Print header
-    print(
-        f"{'ID':<30} {'Size (Metadata)':<20} {'Format':<10} {'Key Span':<10} "
-        f"{'Level':<5} {'Committed At':<20} {'Compacted At':<20}"
-    )
-    print("=" * 120)
+    # Create rich table
+    table = Table(title=None, show_header=True, header_style="bold")
+    table.add_column("ID", style="cyan", no_wrap=True)
+    table.add_column("Size (Metadata)", justify="right")
+    table.add_column("Format", justify="center")
+    table.add_column("Key Span", justify="center")
+    table.add_column("Level", justify="center")
+    table.add_column("Committed At", justify="center")
+    table.add_column("Compacted At", justify="center")
 
-    # Print each fragment
+    # Add each fragment as a row
     for fragment in manifest:
         committed_str = str(datetime_.from_timestamp_micros(fragment.committed_at)) if fragment.committed_at else "N/A"
         compacted_str = str(datetime_.from_timestamp_micros(fragment.compacted_at)) if fragment.compacted_at else "N/A"
@@ -68,12 +74,14 @@ def _table_of_fragments(manifest: FragmentManifest, title: str):
         )
         key_span = f"{fragment.key_span.begin}..{fragment.key_span.end}"
 
-        print(
-            f"{fragment.id:<30} "
-            f"{size_with_metadata:<20} "
-            f"{str(fragment.format):<10} "
-            f"{key_span:<10} "
-            f"{str(fragment.level):<5} "
-            f"{committed_str:<20} "
-            f"{compacted_str:<20}"
+        table.add_row(
+            fragment.id,
+            size_with_metadata,
+            str(fragment.format),
+            key_span,
+            str(fragment.level),
+            committed_str,
+            compacted_str,
         )
+
+    console.print(table)

spiral/expressions/__init__.py

@@ -92,7 +92,7 @@ def lift(expr: ExprLike) -> Expr:
         return lift(pa.array(expr))
 
     # Unpack tables and chunked arrays
-    if isinstance(expr, pa.Table):
+    if isinstance(expr, pa.Table | pa.RecordBatch):
         expr = expr.to_struct_array()
     if isinstance(expr, pa.ChunkedArray):
         expr = expr.combine_chunks()
@@ -104,7 +104,7 @@ def lift(expr: ExprLike) -> Expr:
         if isinstance(expr, pa.StructArray) and expr.null_count != 0:
             # raise ValueError("lift: cannot lift a struct array with nulls.")
             warnings.warn("found a struct array with nulls", stacklevel=2)
-        if isinstance(expr, pa.StructScalar) and not expr.is_valid():
+        if isinstance(expr, pa.StructScalar) and not expr.is_valid:
             # raise ValueError("lift: cannot lift a struct scalar with nulls.")
             warnings.warn("found a struct scalar with nulls", stacklevel=2)
         return lift(arrow_.nest_structs(expr))

spiral/expressions/base.py

@@ -26,14 +26,16 @@ class Expr:
     def __repr__(self):
         return repr(self.__expr__)
 
-    def __getitem__(self, item: str | int) -> "Expr":
+    def __getitem__(self, item: str | int | list[str]) -> "Expr":
         """
         Get an item from a struct or list.
 
         Args:
             item: The key or index to get.
                 If item is a string, it is assumed to be a field in a struct. Dot-separated string is supported
-                to access nested fields. If item is an integer, it is assumed to be an index in a list.
+                to access nested fields.
+                If item is a list of strings, it is assumed to be a list of fields in a struct.
+                If item is an integer, it is assumed to be an index in a list.
         """
         from spiral import expressions as se
 
@@ -42,10 +44,14 @@ class Expr:
         if isinstance(item, int):
             # Assume list and get an element.
             expr = se.list_.element_at(expr, item)
-        else:
+        elif isinstance(item, str):
             # Walk into the struct.
             for part in item.split("."):
                 expr = se.getitem(expr, part)
+        elif isinstance(item, list) and all(isinstance(i, str) for i in item):
+            expr = se.pack({k: expr[k] for k in item})
+        else:
+            raise TypeError(f"Invalid item type: {type(item)}")
 
         return expr
 

spiral/iterable_dataset.py

@@ -0,0 +1,106 @@
+from collections.abc import Callable, Iterator
+from typing import TYPE_CHECKING
+
+import pyarrow as pa
+
+if TYPE_CHECKING:
+    import datasets.iterable_dataset as hf  # noqa
+    import streaming  # noqa
+    import torch.utils.data as torchdata  # noqa
+
+
+def _hf_compatible_schema(schema: pa.Schema) -> pa.Schema:
+    """
+    Replace string-view and binary-view columns in the schema with strings/binary.
+    Recursively handles nested types (struct, list, etc).
+    We use this converted schema as Features in the returned Dataset.
+    Remove this method once we have https://github.com/huggingface/datasets/pull/7718
+    """
+
+    def _convert_type(dtype: pa.DataType) -> pa.DataType:
+        if dtype == pa.string_view():
+            return pa.string()
+        elif dtype == pa.binary_view():
+            return pa.binary()
+        elif pa.types.is_struct(dtype):
+            new_fields = [
+                pa.field(field.name, _convert_type(field.type), nullable=field.nullable, metadata=field.metadata)
+                for field in dtype
+            ]
+            return pa.struct(new_fields)
+        elif pa.types.is_list(dtype):
+            return pa.list_(_convert_type(dtype.value_type))
+        elif pa.types.is_large_list(dtype):
+            return pa.large_list(_convert_type(dtype.value_type))
+        elif pa.types.is_fixed_size_list(dtype):
+            return pa.list_(_convert_type(dtype.value_type), dtype.list_size)
+        elif pa.types.is_map(dtype):
+            return pa.map_(_convert_type(dtype.key_type), _convert_type(dtype.item_type))
+        else:
+            return dtype
+
+    new_fields = []
+    for field in schema:
+        new_type = _convert_type(field.type)
+        new_fields.append(pa.field(field.name, new_type, nullable=field.nullable, metadata=field.metadata))
+
+    return pa.schema(new_fields)
+
+
+def to_iterable_dataset(stream: pa.RecordBatchReader) -> "hf.IterableDataset":
+    from datasets import DatasetInfo, Features
+    from datasets.builder import ArrowExamplesIterable
+    from datasets.iterable_dataset import IterableDataset
+
+    def _generate_tables(**kwargs) -> Iterator[tuple[int, pa.Table]]:
+        # This key is unused when training with IterableDataset.
+        # Default implementation returns shard id, e.g. parquet row group id.
+        for i, rb in enumerate(stream):
+            yield i, pa.Table.from_batches([rb], stream.schema)
+
+    # TODO(marko): This is temporary until we stop returning IterableDataset from this function.
+    class _IterableDataset(IterableDataset):
+        # Diff with datasets.iterable_dataset.IterableDataset:
+        # - Removes torch handling which attempts to handle worker processes.
+        # - Assumes arrow iterator.
+        def __iter__(self):
+            from datasets.formatting import get_formatter
+
+            prepared_ex_iterable = self._prepare_ex_iterable_for_iteration()
+            if self._formatting and (prepared_ex_iterable.iter_arrow or self._formatting.is_table):
+                formatter = get_formatter(self._formatting.format_type, features=self.features)
+                iterator = prepared_ex_iterable.iter_arrow()
+                for key, pa_table in iterator:
+                    yield formatter.format_row(pa_table)
+                return
+
+            for key, example in prepared_ex_iterable:
+                # no need to format thanks to FormattedExamplesIterable
+                yield example
+
+        def map(self, *args, **kwargs):
+            # Map constructs a new IterableDataset, so we need to "patch" it
+            base = super().map(*args, **kwargs)
+            if isinstance(base, IterableDataset):
+                # Patch __iter__ to avoid torch handling
+                base.__class__ = _IterableDataset  # type: ignore
+            return base
+
+    class _ArrowExamplesIterable(ArrowExamplesIterable):
+        def __init__(self, generate_tables_fn: Callable[..., Iterator[tuple[int, pa.Table]]], features: Features):
+            # NOTE: generate_tables_fn type annotations are wrong, return type must be an iterable of tuples.
+            super().__init__(generate_tables_fn, kwargs={})  # type: ignore
+            self._features = features
+
+        @property
+        def is_typed(self) -> bool:
+            return True
+
+        @property
+        def features(self) -> Features:
+            return self._features
+
+    target_features = Features.from_arrow_schema(_hf_compatible_schema(stream.schema))
+    ex_iterable = _ArrowExamplesIterable(_generate_tables, target_features)
+    info = DatasetInfo(features=target_features)
+    return _IterableDataset(ex_iterable=ex_iterable, info=info)

spiral/protogen/_/arrow/flight/protocol/sql/__init__.py

@@ -69,7 +69,7 @@ import betterproto2
 
 from .....message_pool import default_message_pool
 
-_COMPILER_VERSION = "0.8.0"
+_COMPILER_VERSION = "0.9.0"
 betterproto2.check_compiler_version(_COMPILER_VERSION)
 
 

spiral/protogen/_/google/protobuf/__init__.py

@@ -21,12 +21,15 @@ __all__ = (
     "FeatureSet",
     "FeatureSetDefaults",
     "FeatureSetDefaultsFeatureSetEditionDefault",
+    "FeatureSetEnforceNamingStyle",
     "FeatureSetEnumType",
     "FeatureSetFieldPresence",
     "FeatureSetJsonFormat",
     "FeatureSetMessageEncoding",
     "FeatureSetRepeatedFieldEncoding",
     "FeatureSetUtf8Validation",
+    "FeatureSetVisibilityFeature",
+    "FeatureSetVisibilityFeatureDefaultSymbolVisibility",
     "FieldDescriptorProto",
     "FieldDescriptorProtoLabel",
     "FieldDescriptorProtoType",
@@ -54,6 +57,7 @@ __all__ = (
     "ServiceOptions",
     "SourceCodeInfo",
     "SourceCodeInfoLocation",
+    "SymbolVisibility",
     "UninterpretedOption",
     "UninterpretedOptionNamePart",
 )
@@ -66,7 +70,7 @@ import betterproto2
 
 from ...message_pool import default_message_pool
 
-_COMPILER_VERSION = "0.8.0"
+_COMPILER_VERSION = "0.9.0"
 betterproto2.check_compiler_version(_COMPILER_VERSION)
 
 
@@ -174,6 +178,14 @@ class ExtensionRangeOptionsVerificationState(betterproto2.Enum):
     UNVERIFIED = 1
 
 
+class FeatureSetEnforceNamingStyle(betterproto2.Enum):
+    ENFORCE_NAMING_STYLE_UNKNOWN = 0
+
+    STYLE2024 = 1
+
+    STYLE_LEGACY = 2
+
+
 class FeatureSetEnumType(betterproto2.Enum):
     ENUM_TYPE_UNKNOWN = 0
 
@@ -224,6 +236,32 @@ class FeatureSetUtf8Validation(betterproto2.Enum):
     NONE = 3
 
 
+class FeatureSetVisibilityFeatureDefaultSymbolVisibility(betterproto2.Enum):
+    DEFAULT_SYMBOL_VISIBILITY_UNKNOWN = 0
+
+    EXPORT_ALL = 1
+    """
+    Default pre-EDITION_2024, all UNSET visibility are export.
+    """
+
+    EXPORT_TOP_LEVEL = 2
+    """
+    All top-level symbols default to export, nested default to local.
+    """
+
+    LOCAL_ALL = 3
+    """
+    All symbols default to local.
+    """
+
+    STRICT = 4
+    """
+    All symbols local by default. Nested types cannot be exported.
+    With special case caveat for message { enum {} reserved 1 to max; }
+    This is the recommended setting for new protos.
+    """
+
+
 class FieldDescriptorProtoLabel(betterproto2.Enum):
     OPTIONAL = 1
     """
@@ -512,6 +550,22 @@ class MethodOptionsIdempotencyLevel(betterproto2.Enum):
     """
 
 
+class SymbolVisibility(betterproto2.Enum):
+    """
+    Describes the 'visibility' of a symbol with respect to the proto import
+    system. Symbols can only be imported when the visibility rules do not prevent
+    it (ex: local symbols cannot be imported).  Visibility modifiers can only set
+    on `message` and `enum` as they are the only types available to be referenced
+    from other files.
+    """
+
+    VISIBILITY_UNSET = 0
+
+    VISIBILITY_LOCAL = 1
+
+    VISIBILITY_EXPORT = 2
+
+
 @dataclass(eq=False, repr=False)
 class Any(betterproto2.Message):
     """
@@ -744,6 +798,13 @@ class DescriptorProto(betterproto2.Message):
     A given name may only be reserved once.
     """
 
+    visibility: "SymbolVisibility" = betterproto2.field(
+        11, betterproto2.TYPE_ENUM, default_factory=lambda: SymbolVisibility(0)
+    )
+    """
+    Support for `export` and `local` keywords on enums.
+    """
+
 
 default_message_pool.register_message("google.protobuf", "DescriptorProto", DescriptorProto)
 
@@ -835,6 +896,13 @@ class EnumDescriptorProto(betterproto2.Message):
     be reserved once.
     """
 
+    visibility: "SymbolVisibility" = betterproto2.field(
+        6, betterproto2.TYPE_ENUM, default_factory=lambda: SymbolVisibility(0)
+    )
+    """
+    Support for `export` and `local` keywords on enums.
+    """
+
 
 default_message_pool.register_message("google.protobuf", "EnumDescriptorProto", EnumDescriptorProto)
 
@@ -895,6 +963,9 @@ class EnumOptions(betterproto2.Message):
     features: "FeatureSet | None" = betterproto2.field(7, betterproto2.TYPE_MESSAGE, optional=True)
     """
     Any features defined in the specific edition.
+    WARNING: This field should only be used by protobuf plugins or special
+    cases like the proto compiler. Other uses are discouraged and
+    developers should rely on the protoreflect APIs for their client language.
     """
 
     uninterpreted_option: "list[UninterpretedOption]" = betterproto2.field(
@@ -942,6 +1013,9 @@ class EnumValueOptions(betterproto2.Message):
     features: "FeatureSet | None" = betterproto2.field(2, betterproto2.TYPE_MESSAGE, optional=True)
     """
     Any features defined in the specific edition.
+    WARNING: This field should only be used by protobuf plugins or special
+    cases like the proto compiler. Other uses are discouraged and
+    developers should rely on the protoreflect APIs for their client language.
     """
 
     debug_redact: "bool" = betterproto2.field(3, betterproto2.TYPE_BOOL)
@@ -1082,10 +1156,26 @@ class FeatureSet(betterproto2.Message):
         6, betterproto2.TYPE_ENUM, default_factory=lambda: FeatureSetJsonFormat(0)
     )
 
+    enforce_naming_style: "FeatureSetEnforceNamingStyle" = betterproto2.field(
+        7, betterproto2.TYPE_ENUM, default_factory=lambda: FeatureSetEnforceNamingStyle(0)
+    )
+
+    default_symbol_visibility: "FeatureSetVisibilityFeatureDefaultSymbolVisibility" = betterproto2.field(
+        8, betterproto2.TYPE_ENUM, default_factory=lambda: FeatureSetVisibilityFeatureDefaultSymbolVisibility(0)
+    )
+
 
 default_message_pool.register_message("google.protobuf", "FeatureSet", FeatureSet)
 
 
+@dataclass(eq=False, repr=False)
+class FeatureSetVisibilityFeature(betterproto2.Message):
+    pass
+
+
+default_message_pool.register_message("google.protobuf", "FeatureSet.VisibilityFeature", FeatureSetVisibilityFeature)
+
+
 @dataclass(eq=False, repr=False)
 class FeatureSetDefaults(betterproto2.Message):
     """
@@ -1340,6 +1430,9 @@ class FieldOptions(betterproto2.Message):
     features: "FeatureSet | None" = betterproto2.field(21, betterproto2.TYPE_MESSAGE, optional=True)
     """
     Any features defined in the specific edition.
+    WARNING: This field should only be used by protobuf plugins or special
+    cases like the proto compiler. Other uses are discouraged and
+    developers should rely on the protoreflect APIs for their client language.
     """
 
     feature_support: "FieldOptionsFeatureSupport | None" = betterproto2.field(
@@ -1438,6 +1531,12 @@ class FileDescriptorProto(betterproto2.Message):
     For Google-internal migration only. Do not use.
     """
 
+    option_dependency: "list[str]" = betterproto2.field(15, betterproto2.TYPE_STRING, repeated=True)
+    """
+    Names of files imported by this file purely for the purpose of providing
+    option extensions. These are excluded from the dependency list above.
+    """
+
     message_type: "list[DescriptorProto]" = betterproto2.field(4, betterproto2.TYPE_MESSAGE, repeated=True)
     """
     All top-level definitions in this file.
@@ -1465,11 +1564,17 @@ class FileDescriptorProto(betterproto2.Message):
     The supported values are "proto2", "proto3", and "editions".
 
     If `edition` is present, this value must be "editions".
+    WARNING: This field should only be used by protobuf plugins or special
+    cases like the proto compiler. Other uses are discouraged and
+    developers should rely on the protoreflect APIs for their client language.
     """
 
     edition: "Edition" = betterproto2.field(14, betterproto2.TYPE_ENUM, default_factory=lambda: Edition(0))
     """
     The edition of the proto file.
+    WARNING: This field should only be used by protobuf plugins or special
+    cases like the proto compiler. Other uses are discouraged and
+    developers should rely on the protoreflect APIs for their client language.
     """
 
 
@@ -1665,6 +1770,9 @@ class FileOptions(betterproto2.Message):
     features: "FeatureSet | None" = betterproto2.field(50, betterproto2.TYPE_MESSAGE, optional=True)
     """
     Any features defined in the specific edition.
+    WARNING: This field should only be used by protobuf plugins or special
+    cases like the proto compiler. Other uses are discouraged and
+    developers should rely on the protoreflect APIs for their client language.
     """
 
     uninterpreted_option: "list[UninterpretedOption]" = betterproto2.field(
@@ -1817,6 +1925,9 @@ class MessageOptions(betterproto2.Message):
     features: "FeatureSet | None" = betterproto2.field(12, betterproto2.TYPE_MESSAGE, optional=True)
     """
     Any features defined in the specific edition.
+    WARNING: This field should only be used by protobuf plugins or special
+    cases like the proto compiler. Other uses are discouraged and
+    developers should rely on the protoreflect APIs for their client language.
     """
 
     uninterpreted_option: "list[UninterpretedOption]" = betterproto2.field(
@@ -1889,6 +2000,9 @@ class MethodOptions(betterproto2.Message):
     features: "FeatureSet | None" = betterproto2.field(35, betterproto2.TYPE_MESSAGE, optional=True)
     """
     Any features defined in the specific edition.
+    WARNING: This field should only be used by protobuf plugins or special
+    cases like the proto compiler. Other uses are discouraged and
+    developers should rely on the protoreflect APIs for their client language.
     """
 
     uninterpreted_option: "list[UninterpretedOption]" = betterproto2.field(
@@ -1921,6 +2035,9 @@ class OneofOptions(betterproto2.Message):
     features: "FeatureSet | None" = betterproto2.field(1, betterproto2.TYPE_MESSAGE, optional=True)
     """
     Any features defined in the specific edition.
+    WARNING: This field should only be used by protobuf plugins or special
+    cases like the proto compiler. Other uses are discouraged and
+    developers should rely on the protoreflect APIs for their client language.
     """
 
     uninterpreted_option: "list[UninterpretedOption]" = betterproto2.field(
@@ -1955,6 +2072,9 @@ class ServiceOptions(betterproto2.Message):
     features: "FeatureSet | None" = betterproto2.field(34, betterproto2.TYPE_MESSAGE, optional=True)
     """
     Any features defined in the specific edition.
+    WARNING: This field should only be used by protobuf plugins or special
+    cases like the proto compiler. Other uses are discouraged and
+    developers should rely on the protoreflect APIs for their client language.
     """
 
     deprecated: "bool" = betterproto2.field(33, betterproto2.TYPE_BOOL)

spiral/protogen/_/scandal/__init__.py

@@ -25,7 +25,7 @@ import grpc
 
 from ..message_pool import default_message_pool
 
-_COMPILER_VERSION = "0.8.0"
+_COMPILER_VERSION = "0.9.0"
 betterproto2.check_compiler_version(_COMPILER_VERSION)
 
 

spiral/protogen/_/spfs/__init__.py

@@ -16,7 +16,7 @@ import betterproto2
 
 from ..message_pool import default_message_pool
 
-_COMPILER_VERSION = "0.8.0"
+_COMPILER_VERSION = "0.9.0"
 betterproto2.check_compiler_version(_COMPILER_VERSION)
 
 

spiral/protogen/_/spql/__init__.py

@@ -17,7 +17,7 @@ import betterproto2
 
 from ..message_pool import default_message_pool
 
-_COMPILER_VERSION = "0.8.0"
+_COMPILER_VERSION = "0.9.0"
 betterproto2.check_compiler_version(_COMPILER_VERSION)
 
 

spiral/protogen/_/substrait/__init__.py

@@ -246,7 +246,7 @@ import betterproto2
 
 from ..message_pool import default_message_pool
 
-_COMPILER_VERSION = "0.8.0"
+_COMPILER_VERSION = "0.9.0"
 betterproto2.check_compiler_version(_COMPILER_VERSION)
 
 
@@ -2885,7 +2885,7 @@ class ExpressionMaskExpression(betterproto2.Message):
     """
     A reference that takes an existing subtype and selectively removes fields
     from it. For example, one might initially have an inner struct with 100
-    fields but a a particular operation only needs to interact with only 2 of
+    fields but a particular operation only needs to interact with only 2 of
     those 100 fields. In this situation, one would use a mask expression to
     eliminate the 98 fields that are not relevant to the rest of the operation
     pipeline.

spiral/protogen/_/substrait/extensions/__init__.py

@@ -18,7 +18,7 @@ import betterproto2
 
 from ...message_pool import default_message_pool
 
-_COMPILER_VERSION = "0.8.0"
+_COMPILER_VERSION = "0.9.0"
 betterproto2.check_compiler_version(_COMPILER_VERSION)