"""Removing folds with ``n_splits``. Schedule-based ``before`` and ``after``-data.
==================================================================================

Dynamic before/after-ranges, with removed partitions.
"""

import pandas
from rics import configure_stuff
from time_split import log_split_progress, plot, split

configure_stuff(datefmt="")

data = pandas.date_range("2022", "2022-2", freq="15s")
config = dict(
    schedule="0 0 * * MON,FRI",
    before=1,
    after=2,
    available=data,
    n_splits=4,
)

plot(**config, show_removed=True)
# %%
# Any non-zero integer before/after-range may be used. Setting ``show_removed=True`` forces plotting of folds that would
# be silently discarded by the :func:`~time_split.split`-function. Later folds are preferred.

for fold in log_split_progress(split(**config), logger="my-logger"):
    print("Doing work..")