diff --git a/daft/logical/schema.py b/daft/logical/schema.py index 401a235d53..93005ddae7 100644 --- a/daft/logical/schema.py +++ b/daft/logical/schema.py @@ -8,7 +8,7 @@ from daft.daft import read_csv_schema as _read_csv_schema from daft.daft import read_json_schema as _read_json_schema from daft.daft import read_parquet_schema as _read_parquet_schema -from daft.datatype import DataType, TimeUnit +from daft.datatype import DataType, TimeUnit, _ensure_registered_super_ext_type if TYPE_CHECKING: import pyarrow as pa @@ -82,6 +82,7 @@ def to_pyarrow_schema(self) -> pa.Schema: Returns: pa.Schema: PyArrow schema that corresponds to the provided Daft schema """ + _ensure_registered_super_ext_type() return self._schema.to_pyarrow_schema() @classmethod diff --git a/daft/series.py b/daft/series.py index 5cbcfe7ba0..fd85d33f13 100644 --- a/daft/series.py +++ b/daft/series.py @@ -213,6 +213,8 @@ def to_arrow(self) -> pa.Array: """ Convert this Series to an pyarrow array. """ + _ensure_registered_super_ext_type() + dtype = self.datatype() arrow_arr = self._series.to_arrow()