time-split 1.2.1__tar.gz → 1.2.2__tar.gz

{time_split-1.2.1 → time_split-1.2.2}/PKG-INFO

@@ -1,20 +1,18 @@
 Metadata-Version: 2.4
 Name: time-split
-Version: 1.2.1
+Version: 1.2.2
 Summary: Time-based k-fold validation splits for heterogeneous data.
-License: MIT
-License-File: LICENSE.md
 Keywords: hyper,parameter,optimization,cross,validation,time,temporal,heterogenous,data
 Author: Richard Sundqvist
-Author-email: richard.sundqvist@live.se
-Requires-Python: >=3.11
+Author-email: Richard Sundqvist <richard.sundqvist@live.se>
+License-Expression: MIT
+License-File: LICENSE.md
 Classifier: Development Status :: 5 - Production/Stable
 Classifier: Intended Audience :: Developers
 Classifier: Intended Audience :: Science/Research
 Classifier: Operating System :: POSIX :: Linux
 Classifier: Operating System :: MacOS
 Classifier: Operating System :: Microsoft :: Windows
-Classifier: License :: OSI Approved :: MIT License
 Classifier: Natural Language :: English
 Classifier: Typing :: Typed
 Classifier: Programming Language :: Python :: 3
@@ -23,19 +21,20 @@ Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
 Classifier: Programming Language :: Python :: 3.14
-Provides-Extra: app
-Provides-Extra: plotting
-Requires-Dist: click ; extra == "app"
-Requires-Dist: matplotlib ; extra == "plotting"
-Requires-Dist: pandas (>=2.0.3)
-Requires-Dist: rics (>=6.0.1)
-Requires-Dist: seaborn ; extra == "plotting"
-Requires-Dist: time-split-app (>=1.1.0) ; extra == "app"
+Requires-Dist: pandas>=2.0.3
+Requires-Dist: rics>=6.0.1
+Requires-Dist: click ; extra == 'app'
+Requires-Dist: time-split-app>=1.1.0 ; extra == 'app'
+Requires-Dist: matplotlib ; extra == 'plotting'
+Requires-Dist: seaborn ; extra == 'plotting'
+Requires-Python: >=3.11
+Project-URL: homepage, https://github.com/rsundqvist/time-split
+Project-URL: repository, https://github.com/rsundqvist/time-split
+Project-URL: documentation, https://time-split
 Project-URL: Bug Tracker, https://github.com/rsundqvist/time-split/issues
 Project-URL: Changelog, https://github.com/rsundqvist/time-split/blob/master/CHANGELOG.md
-Project-URL: Documentation, https://time-split
-Project-URL: Homepage, https://github.com/rsundqvist/time-split
-Project-URL: Repository, https://github.com/rsundqvist/time-split
+Provides-Extra: app
+Provides-Extra: plotting
 Description-Content-Type: text/markdown
 
 # Time Split  <!-- omit in toc -->
@@ -120,4 +119,3 @@ All contributions, bug reports, bug fixes, documentation improvements, enhanceme
 started, see the [Contributing Guide](CONTRIBUTING.md) and [Code of Conduct](CODE_OF_CONDUCT.md).
 
 [Python Package Index (PyPI)]: https://pypi.org/project/time-split
-

time_split-1.2.2/pyproject.toml

@@ -0,0 +1,203 @@
+[project]
+name = "time-split"
+version = "1.2.2"
+description = "Time-based k-fold validation splits for heterogeneous data."
+license = "MIT"
+license-files = ["LICENSE.md"]
+readme = "README.md"
+keywords = [
+    "hyper",
+    "parameter",
+    "optimization",
+    "cross",
+    "validation",
+    "time",
+    "temporal",
+    "heterogenous",
+    "data",
+]
+classifiers = [
+    "Development Status :: 5 - Production/Stable",
+    "Intended Audience :: Developers",
+    "Intended Audience :: Science/Research",
+    "Operating System :: POSIX :: Linux",
+    "Operating System :: MacOS",
+    "Operating System :: Microsoft :: Windows",
+    "Natural Language :: English",
+    "Typing :: Typed",
+    "Programming Language :: Python :: 3",
+    "Programming Language :: Python :: 3 :: Only",
+    "Programming Language :: Python :: 3.11",
+    "Programming Language :: Python :: 3.12",
+    "Programming Language :: Python :: 3.13",
+    "Programming Language :: Python :: 3.14",
+]
+requires-python = ">=3.11"
+dependencies = [
+    "pandas >= 2.0.3",
+    "rics   >= 6.0.1",
+]
+
+[[project.authors]]
+name = "Richard Sundqvist"
+email = "richard.sundqvist@live.se"
+
+[project.optional-dependencies]
+app = [
+    "click",
+    "time-split-app >= 1.1.0",
+]
+plotting = [
+    "matplotlib",
+    "seaborn",
+]
+
+[project.urls]
+homepage = "https://github.com/rsundqvist/time-split"
+repository = "https://github.com/rsundqvist/time-split"
+documentation = "https://time-split"
+"Bug Tracker" = "https://github.com/rsundqvist/time-split/issues"
+Changelog = "https://github.com/rsundqvist/time-split/blob/master/CHANGELOG.md"
+
+[project.entry-points."rics.cli"]
+time-split = "time_split.cli:main"
+
+[dependency-groups]
+manual-extras = [
+    "croniter~=6.2.4",
+    "pyarrow~=26.0.0",
+    "polars[pandas]~=2.0.0",
+]
+docs = [
+    "sphinx~=9.1.0 ; python_version >= '3.12'",
+    "sphinx-gallery~=0.22.1 ; python_version >= '3.12'",
+    "pydata-sphinx-theme~=0.23.0 ; python_version >= '3.12'",
+    "myst-parser~=5.1.0 ; python_version >= '3.12'",
+    "nbsphinx~=0.9.8 ; python_version >= '3.12'",
+    "ipython~=9.17.1 ; python_version >= '3.12'",
+    "sphinx-llm~=1.0.0 ; python_version >= '3.12'",
+]
+notebooks = [
+    "jupyterlab~=4.6.4",
+    "ipywidgets",
+    "jupyterlab-execute-time",
+    "tqdm",
+]
+test = [
+    "pytest~=9.1.1",
+    "xdoctest~=1.3.2",
+    "coverage~=7.16.2",
+    "pytest-cov~=7.1.0",
+    "scikit-learn~=1.9.1",
+    "dask[dataframe]~=2026.8.0",
+]
+devops = [
+    "nox~=2026.8.17",
+    "invoke~=3.0.3",
+    "mypy~=2.4.0",
+    "ruff~=0.17.1",
+    "codespell~=2.4.3",
+    "bump2version==1.0.1",
+]
+types = ["types-croniter"]
+
+[tool.uv]
+default-groups = [
+    "manual-extras",
+    "docs",
+    "test",
+    "devops",
+    "types",
+]
+
+[tool.ruff]
+line-length = 120
+fix = true
+unsafe-fixes = false
+output-format = "grouped"
+
+[tool.ruff.format]
+docstring-code-format = true
+docstring-code-line-length = 80
+line-ending = "lf"
+
+[tool.ruff.lint]
+exclude = ["notebooks/*"]
+select = [
+    "ANN",
+    "S",
+    "I",
+    "B",
+    "D",
+    "F",
+    "W",
+    "E",
+    "SLOT",
+    "SIM",
+    "TID",
+    "TD",
+    "TCH",
+    "ARG",
+    "PTH",
+    "NPY",
+    "PD",
+    "N",
+    "PERF",
+    "PLC",
+    "PLE",
+    "PLR",
+    "PLW",
+    "UP",
+    "FURB",
+    "RUF",
+    "TRY",
+]
+ignore = [
+    "D105",
+    "D107",
+    "ANN401",
+    "TID252",
+    "TRY003",
+    "PLC0415",
+    "PLR0913",
+    "SIM108",
+    "TD003",
+    "TC006",
+]
+
+[tool.ruff.lint.per-file-ignores]
+"tests/*" = [
+    "D",
+    "S101",
+    "ANN",
+    "PLR2004",
+]
+"examples/*" = [
+    "D205",
+    "B007",
+]
+"**/time_split/*" = ["PLR0913"]
+"**/time_split/integration/sklearn/*" = [
+    "N803",
+    "ARG002",
+]
+"**/time_split/app/*" = ["S101"]
+
+[tool.ruff.lint.pydocstyle]
+convention = "google"
+
+[tool.ruff.lint.pycodestyle]
+max-line-length = 150
+
+[tool.pytest.ini_options]
+filterwarnings = [
+    "error",
+    "ignore:Matplotlib is currently using agg:UserWarning",
+    "ignore:.*'mode' parameter.*Pillow 13.*:DeprecationWarning",
+]
+log_file_level = 10
+log_file = "pytest.log"
+
+[build-system]
+requires = ["uv_build>=0.13.0,<0.14.0"]
+build-backend = "uv_build"

time_split-1.2.1/pyproject.toml → time_split-1.2.2/pyproject.toml.orig

@@ -1,11 +1,12 @@
 [project]
 name = "time-split"
-version = "1.2.1"
+version = "1.2.2"
 description = "Time-based k-fold validation splits for heterogeneous data."
 authors = [
     { name = "Richard Sundqvist", email = "richard.sundqvist@live.se" },
 ]
-license = { text = "MIT" }
+license = "MIT"
+license-files = ["LICENSE.md"]  # uv_build only ships license files listed here.
 readme = "README.md"
 keywords = ["hyper", "parameter", "optimization", "cross", "validation", "time", "temporal", "heterogenous", "data"]
 classifiers = [
@@ -15,7 +16,6 @@ classifiers = [
     "Operating System :: POSIX :: Linux",
     "Operating System :: MacOS",
     "Operating System :: Microsoft :: Windows",
-    "License :: OSI Approved :: MIT License",
     "Natural Language :: English",
     "Typing :: Typed",
     "Programming Language :: Python :: 3",
@@ -46,53 +46,53 @@ documentation = "https://time-split"
 [project.entry-points."rics.cli"]
 time-split = "time_split.cli:main"
 
-[tool.poetry.requires-plugins]
-poetry-plugin-export = ">=1.8"
-
-[tool.poetry.group.manual-extras.dependencies]
-# Extras that must be available for docs + tests go here. Doc/test-only deps go in their respective groups.
-croniter = "~6.0.0"
-pyarrow = "~22.0.0"
-polars = { version = "~1.35.2", extras = ["pandas"] }
-
-[tool.poetry.group.docs]
-[tool.poetry.group.docs.dependencies]
-# duplicate object warnings for sphinx>=4
-# description: https://github.com/astropy/astropy/issues/11723
-# https://github.com/astropy/astropy/pull/12270
-sphinx = { version = "~9.1.0", python = ">=3.12" }
-sphinx-gallery = { version = "~0.20.0", python = ">=3.12" }
-pydata-sphinx-theme = { version = "~0.16.1", python = ">=3.12" }
-myst-parser = { version = "~5.0.0", python = ">=3.12" } # https://github.com/executablebooks/MyST-Parser/issues/749
-nbsphinx = { version = "~0.9.8", python = ">=3.12" }
-ipython = { version = "~9.10.0", python = ">=3.12" }
-sphinx-llm = { version = "~0.3.0", python = ">=3.12" }
-
-[tool.poetry.group.notebooks]
-optional = true
-[tool.poetry.group.notebooks.dependencies]
-jupyterlab = "~4.5.2"
-ipywidgets = "*"
-jupyterlab-execute-time = "*"
-tqdm = "*"
-
-[tool.poetry.group.test.dependencies]
-pytest = "~9.0.2"
-xdoctest = "~1.3.0"
-coverage = "~7.13.1"
-pytest-cov = "~7.0.0"
-
-scikit-learn = "~1.8.0"
-dask = { version = "~2026.1.1", extras = ["dataframe"] }
-
-[tool.poetry.group.devops.dependencies]
-invoke = "~2.2.1"
-mypy = "~1.19.0"
-ruff = "~0.14.13"
+[dependency-groups]
+manual-extras = [
+    # Extras that must be available for docs + tests go here. Doc/test-only deps go in their respective groups.
+    "croniter~=6.2.4",
+    "pyarrow~=26.0.0",
+    "polars[pandas]~=2.0.0",
+]
+docs = [
+    # duplicate object warnings for sphinx>=4
+    # description: https://github.com/astropy/astropy/issues/11723
+    # https://github.com/astropy/astropy/pull/12270
+    "sphinx~=9.1.0 ; python_version >= '3.12'",
+    "sphinx-gallery~=0.22.1 ; python_version >= '3.12'",
+    "pydata-sphinx-theme~=0.23.0 ; python_version >= '3.12'",
+    "myst-parser~=5.1.0 ; python_version >= '3.12'", # https://github.com/executablebooks/MyST-Parser/issues/749
+    "nbsphinx~=0.9.8 ; python_version >= '3.12'",
+    "ipython~=9.17.1 ; python_version >= '3.12'",
+    "sphinx-llm~=1.0.0 ; python_version >= '3.12'", # 1.1.0 warns "unknown node type" on nbsphinx nodes, failing -W
+]
+notebooks = [
+    "jupyterlab~=4.6.4",
+    "ipywidgets",
+    "jupyterlab-execute-time",
+    "tqdm",
+]
+test = [
+    "pytest~=9.1.1",
+    "xdoctest~=1.3.2",
+    "coverage~=7.16.2",
+    "pytest-cov~=7.1.0",
+    "scikit-learn~=1.9.1",
+    "dask[dataframe]~=2026.8.0",
+]
+devops = [
+    "nox~=2026.8.17",
+    "invoke~=3.0.3",
+    "mypy~=2.4.0",
+    "ruff~=0.17.1",
+    "codespell~=2.4.3",
+    "bump2version==1.0.1",
+]
+types = [
+    "types-croniter",
+]
 
-safety = "~3.7.0"
-codespell = "~2.4.1"
-bump2version = "==1.0.1"
+[tool.uv]
+default-groups = ["manual-extras", "docs", "test", "devops", "types"]  # Not notebooks (opt-in).
 
 [tool.ruff]
 line-length = 120
@@ -188,5 +188,5 @@ log_file_level = 10
 log_file = "pytest.log"
 
 [build-system]
-requires = ["poetry-core>=1.0.8"]
-build-backend = "poetry.core.masonry.api"
+requires = ["uv_build>=0.13.0,<0.14.0"]
+build-backend = "uv_build"

{time_split-1.2.1 → time_split-1.2.2}/src/time_split/__init__.py

@@ -9,4 +9,4 @@ __all__ = [
     "split",
 ]
 
-__version__ = "1.2.1"
+__version__ = "1.2.2"

{time_split-1.2.1 → time_split-1.2.2}/src/time_split/_backend/_limits.py

@@ -165,7 +165,7 @@ def _make_level(
 ) -> _TimedeltaTuple:
     start_at = make_timedelta(start_at)  # Will be NaT if None, making all ifs False.
     if start_at <= Timedelta(0):
-        raise ValueError(f"Bad {start_at=}; must be non-negative.")
+        raise ValueError(f"Bad {start_at=}; must be positive.")
 
     if isinstance(round_to, str):
         try:
@@ -175,7 +175,7 @@ def _make_level(
 
     tolerance = make_timedelta(tolerance)
     if tolerance <= Timedelta(0):
-        raise ValueError(f"Bad {tolerance=}; must be non-negative.")
+        raise ValueError(f"Bad {tolerance=}; must be positive.")
 
     if start_at < round_to:
         raise ValueError(f"{start_at=} < {round_to=}")

{time_split-1.2.1 → time_split-1.2.2}/src/time_split/_backend/_process_available.py

@@ -56,4 +56,4 @@ def process_available(available: DatetimeIterable, *, expand_limits: ExpandLimit
 
 
 def _compute_data_limits(available: DatetimeIndexLike) -> LimitsTuple:
-    return handle_dask(available.min()), handle_dask(available.max())
+    return handle_dask(available.min(), available.max())

{time_split-1.2.1 → time_split-1.2.2}/src/time_split/_backend/_schedule.py

@@ -34,6 +34,9 @@ def materialize_schedule(
         except TypeError as e:
             raise ValueError("Schedule must be explicit when not bounded by an available range.") from e
 
+    if isinstance(schedule, str) and not schedule.strip():
+        raise ValueError(f"Schedule must not be blank: {schedule=}")
+
     available_metadata = process_available(available, expand_limits=expand_limits)
     min_dt, max_dt = available_metadata.expanded_limits
 

{time_split-1.2.1 → time_split-1.2.2}/src/time_split/_compat.py

@@ -1,10 +1,15 @@
+import re
+
 from pandas import Timedelta
 
 from time_split.types import TimedeltaTypes
 
+# TODO(pandas>=4): Remove this module.
+
 
 def fix_pandas4_warning(arg: str) -> str:
-    return arg[:-1] + "D" if arg.endswith("d") else arg
+    # Uppercase a "d" unit that starts the string or follows a number or space ("d", "1d", "7 d", "1d2h"), not "days".
+    return re.sub(r"(?<![^\d\s])d(?![a-z])", "D", arg)
 
 
 def make_timedelta(arg: TimedeltaTypes) -> Timedelta:

{time_split-1.2.1 → time_split-1.2.2}/src/time_split/_frontend/_plot.py

@@ -169,7 +169,7 @@ def plot(
     ax.set_title(_make_title(available, split_kwargs))
 
     if plot_data.available is None:
-        ax.legend(loc=("lower" if splitter.step > 0 else "upper") + " right")
+        ax.legend(loc="lower right" if splitter.step > 0 else "upper right")
         return ax
 
     _plot_limits(ax, plot_data.available.expanded_limits)
@@ -178,7 +178,7 @@ def plot(
         assert isinstance(row_count_bin, (str, pd.Series))  # noqa: S101
         _plot_row_counts(ax, row_count_bin, plot_data.row_counts)
 
-    ax.legend(loc=("lower" if splitter.step > 0 else "upper") + " right")
+    ax.legend(loc="lower right" if splitter.step > 0 else "upper right")
 
     return ax
 

{time_split-1.2.1 → time_split-1.2.2}/src/time_split/_frontend/_progress.py

@@ -93,9 +93,9 @@ def log_split_progress(
     """
     logger = logging.getLogger(logger) if isinstance(logger, str) else logger
 
-    if isinstance(logger, logging.LoggerAdapter) and not hasattr(logger, "merge_extra"):
-        # Backport of https://github.com/python/cpython/pull/107292
-        logger = _MergingLoggerAdapter(logger.logger, logger.extra)
+    if isinstance(logger, logging.LoggerAdapter) and not getattr(logger, "merge_extra", False):
+        # Stock adapters drop per-call extras (https://github.com/python/cpython/pull/107292); wrap to merge them.
+        logger = _MergingLoggerAdapter(logger)
 
     return _ProgressTracker(
         splits,
@@ -218,7 +218,7 @@ def default_metrics_formatter(end_message: str, metrics: dict[Any, Any] | pd.Ser
 
     if isinstance(metrics, (pd.Series, pd.DataFrame)):
         if isinstance(metrics, pd.DataFrame):
-            with_index = not isinstance(metrics, pd.RangeIndex)
+            with_index = not isinstance(metrics.index, pd.RangeIndex)
         else:
             with_index = True
         return f"{end_message} Fold metrics:\n{metrics.round(3).to_string(index=with_index)}"
@@ -229,10 +229,17 @@ def default_metrics_formatter(end_message: str, metrics: dict[Any, Any] | pd.Ser
 
 
 class _MergingLoggerAdapter(logging.LoggerAdapter[Any]):
-    # TODO(3.13): Use merge_extra=True init arg
+    def __init__(self, adapter: logging.LoggerAdapter[Any]) -> None:
+        super().__init__(adapter.logger, adapter.extra)
+        self._adapter = adapter
+
     def process(self, msg: Any, kwargs: MutableMapping[str, Any]) -> tuple[Any, MutableMapping[str, Any]]:
         """See https://github.com/python/cpython/pull/107292."""
-        kwargs["extra"] = {**self.extra, **kwargs["extra"]} if "extra" in kwargs and self.extra else self.extra
+        call_extra = kwargs.get("extra") or {}
+        # Let the wrapped adapter (possibly a user subclass) transform msg/kwargs first.
+        msg, kwargs = self._adapter.process(msg, kwargs)
+        adapter_extra = kwargs.get("extra") or {}
+        kwargs["extra"] = {**adapter_extra, **call_extra}
         return msg, kwargs
 
 

{time_split-1.2.1 → time_split-1.2.2}/src/time_split/_frontend/_weight.py

@@ -54,14 +54,10 @@ def _from_data(available: DatetimeIterable, splits: DatetimeSplits) -> list[Date
     else:
         time = DatetimeIndex(available).to_series()  # Coerce everything else.
 
-    retval = []
-    for fold in splits:
-        data = time.between(fold.start, fold.mid, inclusive="left").sum()
-        future_data = time.between(fold.mid, fold.end, inclusive="left").sum()
-
-        data = handle_dask(data)
-        future_data = handle_dask(future_data)
-
-        retval.append(DatetimeSplitCounts(data, future_data=future_data))
-
-    return retval
+    counts = [
+        time.between(a, b, inclusive="left").sum()
+        for fold in splits
+        for a, b in ((fold.start, fold.mid), (fold.mid, fold.end))
+    ]
+    computed = handle_dask(*counts)
+    return [DatetimeSplitCounts(d, future_data=fd) for d, fd in zip(computed[::2], computed[1::2], strict=True)]

time_split-1.2.2/src/time_split/_support.py

@@ -0,0 +1,26 @@
+from typing import Any, Protocol, TypeVar, overload, runtime_checkable
+
+T_co = TypeVar("T_co", covariant=True)
+
+
+@runtime_checkable
+class _Computable(Protocol[T_co]):
+    def compute(self) -> T_co:
+        """Compute results of a delayed Dask object."""
+
+
+@overload
+def handle_dask(arg: T_co | _Computable[T_co], /) -> T_co: ...
+@overload
+def handle_dask(a: T_co | _Computable[T_co], b: T_co | _Computable[T_co], /) -> tuple[T_co, T_co]: ...
+@overload
+def handle_dask(*args: T_co | _Computable[T_co]) -> tuple[T_co, ...]: ...
+def handle_dask(*args: Any) -> Any:
+    """Run Dask compute on delayed objects, in a single pass."""
+    if any(isinstance(a, _Computable) for a in args):
+        # shortcut: non-Dask computables (e.g. xarray) now require dask; accepted, add a .compute() fallback if reported.
+        from dask import compute  # type: ignore[attr-defined]
+
+        args = compute(*args)  # type: ignore[no-untyped-call]
+
+    return args[0] if len(args) == 1 else args

{time_split-1.2.1 → time_split-1.2.2}/src/time_split/integration/pandas/_impl.py

@@ -2,7 +2,7 @@ from collections.abc import Hashable, Iterable
 from datetime import date, datetime
 from typing import Generic, TypeVar, Unpack
 
-from pandas import DataFrame, DatetimeIndex, Series, Timestamp
+from pandas import DataFrame, DatetimeIndex, Index, Series, Timestamp
 from rics.misc import tname
 
 from ..._docstrings import docs
@@ -60,6 +60,9 @@ class _Indexer(Generic[PandasT]):
     def as_available(self, data: PandasT) -> Series | DatetimeIndex:
         time = self._get_time(data)
 
+        if time.empty:
+            return time
+
         first = time.iloc[0] if hasattr(time, "iloc") else time[0]
         if isinstance(first, date):
             return time
@@ -74,8 +77,12 @@ class _Indexer(Generic[PandasT]):
     def select(self, data: PandasT, left: datetime, right: datetime) -> PandasT:
         """Select data based on the given bounds."""
         time = self._get_time(data)
-        if isinstance(time, Series):
-            return data[time.between(left, right, inclusive="left")]
-        else:
-            # Index slicing is a lot faster than boolean masks (empirically, seems to be a factor ~10).
-            return data[left : right - Timestamp.resolution]  # type: ignore[misc]
+
+        if isinstance(time, Index):
+            if time.is_monotonic_increasing:
+                # Index slicing is a lot faster than boolean masks (empirically, seems to be a factor ~10).
+                return data[left : right - Timestamp.resolution]
+
+            time = time.to_series()
+
+        return data[time.between(left, right, inclusive="left")]

{time_split-1.2.1 → time_split-1.2.2}/src/time_split/integration/polars/_impl.py

@@ -54,7 +54,7 @@ class _Indexer:
     def as_available(self, data: DataFrame) -> Series:
         time = self._get_time(data)
 
-        if isinstance(time[0], date):
+        if time.is_empty() or isinstance(time[0], date):
             return time
 
         msg = f"Elements of data[{self.time_column!r}] element are {tname(time[0])}, expected datetime-like."

time_split-1.2.1/src/time_split/_support.py

@@ -1,14 +0,0 @@
-from typing import Protocol, TypeVar, runtime_checkable
-
-T_co = TypeVar("T_co", covariant=True)
-
-
-@runtime_checkable
-class _Computable(Protocol[T_co]):
-    def compute(self) -> T_co:
-        """Compute results of a delayed Dask object."""
-
-
-def handle_dask(arg: T_co | _Computable[T_co]) -> T_co:
-    """Run Dask compute function on a delayed object."""
-    return arg.compute() if isinstance(arg, _Computable) else arg