diff --git a/python/cudf_polars/cudf_polars/experimental/__init__.py b/python/cudf_polars/cudf_polars/experimental/__init__.py index 4a2aef68e18..6fd93bf5157 100644 --- a/python/cudf_polars/cudf_polars/experimental/__init__.py +++ b/python/cudf_polars/cudf_polars/experimental/__init__.py @@ -6,8 +6,3 @@ from __future__ import annotations __all__: list[str] = [] - -from cudf_polars.experimental.parallel import _register - -# Register multi-partition dispatch functions -_register() diff --git a/python/cudf_polars/cudf_polars/experimental/parallel.py b/python/cudf_polars/cudf_polars/experimental/parallel.py index 091ba6d0791..a783f91fc73 100644 --- a/python/cudf_polars/cudf_polars/experimental/parallel.py +++ b/python/cudf_polars/cudf_polars/experimental/parallel.py @@ -307,6 +307,9 @@ def _( ## -def _register(): +def _register_other(): """Register multi-partition dispatch functions.""" import cudf_polars.experimental.io # noqa: F401 + + +_register_other()