time-split 1.0.4__tar.gz → 1.1.0__tar.gz

{time_split-1.0.4 → time_split-1.1.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: time-split
-Version: 1.0.4
+Version: 1.1.0
 Summary: Time-based k-fold validation splits for heterogeneous data.
 License: MIT
 License-File: LICENSE.md

{time_split-1.0.4 → time_split-1.1.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "time-split"
-version = "1.0.4"
+version = "1.1.0"
 description = "Time-based k-fold validation splits for heterogeneous data."
 authors = [
     { name = "Richard Sundqvist", email = "richard.sundqvist@live.se" },
@@ -53,7 +53,7 @@ poetry-plugin-export = ">=1.8"
 # 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.34.0", extras = ["pandas"] }
+polars = { version = "~1.35.2", extras = ["pandas"] }
 
 [tool.poetry.group.docs]
 [tool.poetry.group.docs.dependencies]
@@ -61,10 +61,10 @@ polars = { version = "~1.34.0", extras = ["pandas"] }
 # description: https://github.com/astropy/astropy/issues/11723
 # https://github.com/astropy/astropy/pull/12270
 sphinx = "~8.1.3" # 8.2 breaks xdoc refs in README.md; maybe https://github.com/sphinx-doc/sphinx/issues/13932?
-sphinx-gallery = "~0.19.0"
+sphinx-gallery = "~0.20.0"
 pydata-sphinx-theme = "~0.16.1"
 myst-parser = "~4.0.1"
-ipython = "~9.6.0"
+ipython = "~9.8.0"
 
 [tool.poetry.group.notebooks]
 optional = true
@@ -75,9 +75,9 @@ jupyterlab-execute-time = "*"
 tqdm = "*"
 
 [tool.poetry.group.test.dependencies]
-pytest = "~8.4.2"
+pytest = "~9.0.1"
 xdoctest = "~1.3.0"
-coverage = "~7.11.0"
+coverage = "~7.12.0"
 pytest-cov = "~7.0.0"
 
 scikit-learn = "~1.7.2"
@@ -85,10 +85,10 @@ dask = { version = "~2025.10.0", extras = ["dataframe"] }
 
 [tool.poetry.group.devops.dependencies]
 invoke = "~2.2.1"
-mypy = "~1.18.2"
-ruff = "~0.14.2"
+mypy = "~1.19.0"
+ruff = "~0.14.8"
 
-safety = "~3.6.2"
+safety = "~3.7.0"
 codespell = "~2.4.1"
 bump2version = "~1.0.1"
 

{time_split-1.0.4 → time_split-1.1.0}/src/time_split/__init__.py

@@ -9,4 +9,4 @@ __all__ = [
     "split",
 ]
 
-__version__ = "1.0.4"
+__version__ = "1.1.0"

{time_split-1.0.4 → time_split-1.1.0}/src/time_split/_backend/_schedule.py

@@ -2,6 +2,7 @@ from typing import NamedTuple, get_args
 
 from pandas import DatetimeIndex, NaT, Timedelta, Timestamp, date_range
 
+from ..settings import misc as settings
 from ..types import DatetimeIterable, ExpandLimits, Schedule, TimedeltaTypes
 from ._limits import LimitsTuple
 from ._process_available import ProcessAvailableResult, process_available
@@ -46,7 +47,7 @@ def materialize_schedule(
 
 
 def _cron_like(schedule: str) -> bool:
-    # CroniterBadCronError: Exactly 5 or 6 columns has to be specified for iterator expression.
+    # CroniterBadCronError: Exactly 5 or 6 columns have to be specified for iterator expression.
     return len(schedule.split()) > 4 or schedule[0] == "@"  # noqa: PLR2004
 
 
@@ -54,7 +55,11 @@ def _from_timedelta(schedule: TimedeltaTypes, limits: LimitsTuple) -> DatetimeIn
     timedelta = Timedelta(schedule)
     if timedelta <= Timedelta(0):
         raise ValueError(f"unbounded {schedule=} must be greater than zero.")
-    return date_range(*limits, freq=timedelta, inclusive="both")
+
+    if settings.snap_to_end:
+        return date_range(limits[1], limits[0], freq=-timedelta, inclusive="both")[::-1]
+    else:
+        return date_range(limits[0], limits[1], freq=timedelta, inclusive="both")
 
 
 def _handle_cron(expr: str, min_dt: Timestamp, max_dt: Timestamp) -> DatetimeIndex:

{time_split-1.0.4 → time_split-1.1.0}/src/time_split/_backend/_splitter.py

@@ -54,19 +54,20 @@ class DatetimeIndexSplitter:
             settings.snap_to_end
             and self.after != "all"
             and isinstance(self.schedule, types)
-            and isinstance(self.after, types)
+            and isinstance(self.after, (types, int))
         ):
             ms = self._snap_to_end(ms)
 
         return ms
 
     def _snap_to_end(self, ms: MaterializedSchedule) -> MaterializedSchedule:
-        schedule_frequency = ms.schedule.freq
-        if schedule_frequency is None:
-            return ms
+        data_end = ms.available_metadata.expanded_limits[1]
+        schedule_end = ms.schedule[-1]
 
-        from_end = ms.available_metadata.expanded_limits[1] - (ms.schedule[-1] + Timedelta(self.after))
-        from_end = from_end.floor(schedule_frequency.base)  # TODO(settings): should this depend on misc.round_limits?
+        from_end = data_end - schedule_end
+        if not isinstance(self.after, int):
+            after = Timedelta(self.after)
+            from_end -= after
 
         return ms._replace(schedule=ms.schedule + from_end)
 

{time_split-1.0.4 → time_split-1.1.0}/src/time_split/app/_create_explorer_link.py

@@ -99,8 +99,8 @@ def _create_query(
         query = {"data": data, **kwargs}
     else:
         splitter = DatetimeIndexSplitter(**kwargs)
-        _, ms = splitter.get_plot_data(data)
-        query["data"] = _get_data_param(*ms.available_metadata.limits)
+        start, end = splitter._materialize_schedule(data).available_metadata.limits
+        query["data"] = _get_data_param(start, end)
 
         if skip_default:
             schedule = kwargs["schedule"]