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

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

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

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

@@ -1,6 +1,6 @@
 [project]
 name = "time-split"
-version = "1.1.0"
+version = "1.1.1"
 description = "Time-based k-fold validation splits for heterogeneous data."
 authors = [
     { name = "Richard Sundqvist", email = "richard.sundqvist@live.se" },

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

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

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

@@ -1,4 +1,4 @@
-from typing import NamedTuple, get_args
+from typing import Literal, NamedTuple, get_args
 
 from pandas import DatetimeIndex, NaT, Timedelta, Timestamp, date_range
 
@@ -8,6 +8,7 @@ from ._limits import LimitsTuple
 from ._process_available import ProcessAvailableResult, process_available
 
 NO_LIMITS: LimitsTuple = NaT, NaT
+ScheduleType = Literal["cron", "explicit", "timedelta"]
 
 
 class MaterializedSchedule(NamedTuple):
@@ -15,6 +16,7 @@ class MaterializedSchedule(NamedTuple):
 
     schedule: DatetimeIndex
     available_metadata: ProcessAvailableResult
+    schedule_type: ScheduleType
 
 
 def materialize_schedule(
@@ -26,6 +28,7 @@ def materialize_schedule(
             return MaterializedSchedule(
                 DatetimeIndex(schedule),
                 available_metadata=ProcessAvailableResult(None, NO_LIMITS, NO_LIMITS),
+                schedule_type="explicit",
             )
         except TypeError as e:
             raise ValueError("Schedule must be explicit when not bounded by an available range.") from e
@@ -33,17 +36,22 @@ def materialize_schedule(
     available_metadata = process_available(available, expand_limits=expand_limits)
     min_dt, max_dt = available_metadata.expanded_limits
 
+    schedule_type: ScheduleType
     if isinstance(schedule, str) and _cron_like(schedule):
         schedule = _handle_cron(schedule, min_dt, max_dt)
+        schedule_type = "cron"
     elif isinstance(schedule, get_args(TimedeltaTypes)):
         schedule = _from_timedelta(schedule, available_metadata.expanded_limits)
+        schedule_type = "timedelta"
     else:
         if not isinstance(schedule, DatetimeIndex):
             schedule = DatetimeIndex(schedule)
 
         schedule = schedule.tz_localize(min_dt.tz) if schedule.tz is None else schedule.tz_convert(min_dt.tz)
         schedule = schedule[(min_dt <= schedule) & (schedule <= max_dt)]
-    return MaterializedSchedule(schedule, available_metadata=available_metadata)
+        schedule_type = "explicit"
+
+    return MaterializedSchedule(schedule, available_metadata, schedule_type)
 
 
 def _cron_like(schedule: str) -> bool:

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

@@ -53,7 +53,7 @@ class DatetimeIndexSplitter:
         if (
             settings.snap_to_end
             and self.after != "all"
-            and isinstance(self.schedule, types)
+            and ms.schedule_type == "timedelta"
             and isinstance(self.after, (types, int))
         ):
             ms = self._snap_to_end(ms)