spotforecast2 0.0.2__tar.gz → 0.0.4__tar.gz

spotforecast2-0.0.4/PKG-INFO

@@ -0,0 +1,77 @@
+Metadata-Version: 2.3
+Name: spotforecast2
+Version: 0.0.4
+Summary: Forecasting with spot
+Author: bartzbeielstein
+Author-email: bartzbeielstein <32470350+bartzbeielstein@users.noreply.github.com>
+Requires-Dist: astral>=3.2
+Requires-Dist: feature-engine>=1.9.3
+Requires-Dist: flake8>=7.3.0
+Requires-Dist: holidays>=0.90
+Requires-Dist: ipykernel>=7.1.0
+Requires-Dist: jupyter>=1.1.1
+Requires-Dist: lightgbm>=4.6.0
+Requires-Dist: matplotlib>=3.10.8
+Requires-Dist: numba>=0.63.1
+Requires-Dist: optuna>=4.7.0
+Requires-Dist: pandas>=3.0.0
+Requires-Dist: plotly>=6.5.2
+Requires-Dist: pyarrow>=23.0.0
+Requires-Dist: scikit-learn>=1.8.0
+Requires-Dist: spotoptim>=0.0.160
+Requires-Dist: tqdm>=4.67.2
+Requires-Dist: pytest>=9.0.2 ; extra == 'dev'
+Requires-Dist: pytest-cov>=6.0.0 ; extra == 'dev'
+Requires-Dist: black>=24.1.0 ; extra == 'dev'
+Requires-Dist: isort>=5.13.0 ; extra == 'dev'
+Requires-Dist: ruff>=0.3.0 ; extra == 'dev'
+Requires-Dist: mkdocs>=1.6.1 ; extra == 'dev'
+Requires-Dist: mkdocs-material>=9.7.1 ; extra == 'dev'
+Requires-Dist: mkdocstrings>=1.0.2 ; extra == 'dev'
+Requires-Dist: mkdocstrings-python>=2.0.1 ; extra == 'dev'
+Requires-Dist: safety>=3.0.0 ; extra == 'dev'
+Requires-Dist: bandit>=1.8.0 ; extra == 'dev'
+Requires-Python: >=3.13
+Provides-Extra: dev
+Description-Content-Type: text/markdown
+
+ # spotforecast2
+
+[![Python Version](https://img.shields.io/badge/python-3.13%2B-blue)](https://www.python.org/downloads/)
+[![PyPI Version](https://img.shields.io/pypi/v/spotforecast2)](https://pypi.org/project/spotforecast2/)
+[![PyPI Downloads](https://img.shields.io/pypi/dm/spotforecast2)](https://pypi.org/project/spotforecast2/)
+[![License](https://img.shields.io/badge/license-BSD--3--Clause-blue)](https://github.com/sequential-parameter-optimization/spotforecast2/blob/main/LICENSE)
+
+**Testing & Quality**
+
+[![Build Status](https://img.shields.io/github/actions/workflow/status/sequential-parameter-optimization/spotforecast2/ci.yml?branch=main&label=Tests)](https://github.com/sequential-parameter-optimization/spotforecast2/actions/workflows/ci.yml)
+[![Documentation](https://img.shields.io/badge/docs-passing-brightgreen)](https://sequential-parameter-optimization.github.io/spotforecast2/)
+[![GitHub Release](https://img.shields.io/github/v/release/sequential-parameter-optimization/spotforecast2)](https://github.com/sequential-parameter-optimization/spotforecast2/releases)
+
+**Status**
+
+[![Maintenance](https://img.shields.io/badge/maintenance-active-green)](https://github.com/sequential-parameter-optimization/spotforecast2)
+[![Code style: black](https://img.shields.io/badge/code%20style-black-000000.svg)](https://github.com/psf/black)
+
+## About spotforecast2
+
+`spotforecast2` is a Python library for time series forecasting. 
+
+Parts of the code are ported from `skforecast` to reduce external dependencies.
+Many thanks to the [skforecast team](https://skforecast.org/0.20.0/more/about-skforecast.html) for their great work!
+
+
+# License
+
+`spotforecast2` software: [BSD-3-Clause License](https://github.com/sequential-parameter-optimization/spotforecast2?tab=BSD-3-Clause-1-ov-file)
+
+
+# References
+
+## skforecast: 
+
+* Amat Rodrigo, J., & Escobar Ortiz, J. (2026). skforecast (Version 0.20.0) [Computer software]. https://doi.org/10.5281/zenodo.8382788 
+
+## spotoptim:
+
+* [spotoptim documentation](https://sequential-parameter-optimization.github.io/spotoptim/)

spotforecast2-0.0.4/README.md

@@ -0,0 +1,40 @@
+ # spotforecast2
+
+[![Python Version](https://img.shields.io/badge/python-3.13%2B-blue)](https://www.python.org/downloads/)
+[![PyPI Version](https://img.shields.io/pypi/v/spotforecast2)](https://pypi.org/project/spotforecast2/)
+[![PyPI Downloads](https://img.shields.io/pypi/dm/spotforecast2)](https://pypi.org/project/spotforecast2/)
+[![License](https://img.shields.io/badge/license-BSD--3--Clause-blue)](https://github.com/sequential-parameter-optimization/spotforecast2/blob/main/LICENSE)
+
+**Testing & Quality**
+
+[![Build Status](https://img.shields.io/github/actions/workflow/status/sequential-parameter-optimization/spotforecast2/ci.yml?branch=main&label=Tests)](https://github.com/sequential-parameter-optimization/spotforecast2/actions/workflows/ci.yml)
+[![Documentation](https://img.shields.io/badge/docs-passing-brightgreen)](https://sequential-parameter-optimization.github.io/spotforecast2/)
+[![GitHub Release](https://img.shields.io/github/v/release/sequential-parameter-optimization/spotforecast2)](https://github.com/sequential-parameter-optimization/spotforecast2/releases)
+
+**Status**
+
+[![Maintenance](https://img.shields.io/badge/maintenance-active-green)](https://github.com/sequential-parameter-optimization/spotforecast2)
+[![Code style: black](https://img.shields.io/badge/code%20style-black-000000.svg)](https://github.com/psf/black)
+
+## About spotforecast2
+
+`spotforecast2` is a Python library for time series forecasting. 
+
+Parts of the code are ported from `skforecast` to reduce external dependencies.
+Many thanks to the [skforecast team](https://skforecast.org/0.20.0/more/about-skforecast.html) for their great work!
+
+
+# License
+
+`spotforecast2` software: [BSD-3-Clause License](https://github.com/sequential-parameter-optimization/spotforecast2?tab=BSD-3-Clause-1-ov-file)
+
+
+# References
+
+## skforecast: 
+
+* Amat Rodrigo, J., & Escobar Ortiz, J. (2026). skforecast (Version 0.20.0) [Computer software]. https://doi.org/10.5281/zenodo.8382788 
+
+## spotoptim:
+
+* [spotoptim documentation](https://sequential-parameter-optimization.github.io/spotoptim/)

{spotforecast2-0.0.2 → spotforecast2-0.0.4}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "spotforecast2"
-version = "0.0.2"
+version = "0.0.4"
 description = "Forecasting with spot"
 readme = "README.md"
 authors = [
@@ -30,11 +30,17 @@ dependencies = [
 requires = ["uv_build>=0.9.18,<0.10.0"]
 build-backend = "uv_build"
 
-[dependency-groups]
+[project.optional-dependencies]
 dev = [
+    "pytest>=9.0.2",
+    "pytest-cov>=6.0.0",
+    "black>=24.1.0",
+    "isort>=5.13.0",
+    "ruff>=0.3.0",
     "mkdocs>=1.6.1",
     "mkdocs-material>=9.7.1",
     "mkdocstrings>=1.0.2",
     "mkdocstrings-python>=2.0.1",
-    "pytest>=9.0.2",
+    "safety>=3.0.0",
+    "bandit>=1.8.0",
 ]

{spotforecast2-0.0.2 → spotforecast2-0.0.4}/src/spotforecast2/forecaster/utils.py

@@ -20,6 +20,11 @@ from spotforecast2.utils import (
 )
 from spotforecast2.exceptions import set_skforecast_warnings, UnknownLevelWarning
 
+try:
+    from tqdm.auto import tqdm
+except ImportError:  # pragma: no cover - fallback when tqdm is not installed
+    tqdm = None
+
 
 def check_preprocess_series(series):
     pass
@@ -785,6 +790,7 @@ def predict_multivariate(
     forecasters: dict[str, Any],
     steps_ahead: int,
     exog: pd.DataFrame | None = None,
+    show_progress: bool = False,
 ) -> pd.DataFrame:
     """
     Generate multi-output predictions using multiple baseline forecasters.
@@ -796,6 +802,8 @@ def predict_multivariate(
         steps_ahead (int): Number of steps to forecast.
         exog (pd.DataFrame, optional): Exogenous variables for prediction.
             If provided, will be passed to each forecaster's predict method.
+        show_progress (bool, optional): Show progress bar while predicting
+            per target forecaster. Default: False.
 
     Returns:
         pd.DataFrame: DataFrame with predictions for all targets.
@@ -824,7 +832,15 @@ def predict_multivariate(
 
     predictions = {}
 
-    for target, forecaster in forecasters.items():
+    target_iter = forecasters.items()
+    if show_progress and tqdm is not None:
+        target_iter = tqdm(
+            forecasters.items(),
+            desc="Predicting targets",
+            unit="model",
+        )
+
+    for target, forecaster in target_iter:
         # Generate predictions for this target
         if exog is not None:
             pred = forecaster.predict(steps=steps_ahead, exog=exog)

{spotforecast2-0.0.2 → spotforecast2-0.0.4}/src/spotforecast2/processing/n2n_predict.py

@@ -1,5 +1,5 @@
 import pandas as pd
-from typing import List, Optional, Tuple, Dict, Any
+from typing import List, Optional
 from spotforecast2.forecaster.recursive import ForecasterEquivalentDate
 from spotforecast2.data.fetch_data import fetch_data
 from spotforecast2.preprocessing.curate_data import basic_ts_checks
@@ -8,9 +8,13 @@ from spotforecast2.preprocessing.outlier import mark_outliers
 
 from spotforecast2.preprocessing.split import split_rel_train_val_test
 from spotforecast2.forecaster.utils import predict_multivariate
-from spotforecast2.model_selection import TimeSeriesFold, backtesting_forecaster
 from spotforecast2.preprocessing.curate_data import get_start_end
 
+try:
+    from tqdm.auto import tqdm
+except ImportError:  # pragma: no cover - fallback when tqdm is not installed
+    tqdm = None
+
 
 def n2n_predict(
     columns: Optional[List[str]] = None,
@@ -18,22 +22,22 @@ def n2n_predict(
     contamination: float = 0.01,
     window_size: int = 72,
     verbose: bool = True,
-) -> Tuple[pd.DataFrame, Optional[Dict[str, Any]]]:
+    show_progress: bool = True,
+) -> pd.DataFrame:
     """
     End-to-end prediction function replicating the workflow from 01_base_predictor combined with fetch_data.
 
     Args:
         columns: List of target columns to forecast. If None, uses a default set (defined internally or from data).
-                 Note: fetch_data now supports None to return all columns.
+                 Note: fetch_data supports None to return all columns.
         forecast_horizon: Number of steps to forecast.
         contamination: Contamination factor for outlier detection.
         window_size: Window size for weighting (not fully utilized in main flow but kept for consistency).
         verbose: Whether to print progress logs.
+        show_progress: Show progress bar during training and prediction.
 
     Returns:
-        Tuple containing:
-            - predictions (pd.DataFrame): The multi-output predictions.
-            - metrics (Optional[Dict]): Dictionary containing backtesting metrics if performed.
+        pd.DataFrame: The multi-output predictions.
     """
     if columns is not None:
         TARGET = columns
@@ -95,7 +99,11 @@ def n2n_predict(
 
     baseline_forecasters = {}
 
-    for target in data.columns:
+    target_iter = data.columns
+    if show_progress and tqdm is not None:
+        target_iter = tqdm(data.columns, desc="Training forecasters", unit="model")
+
+    for target in target_iter:
         forecaster = ForecasterEquivalentDate(offset=pd.DateOffset(days=1), n_offsets=1)
 
         forecaster.fit(y=data.loc[:end_validation, target])
@@ -105,13 +113,14 @@ def n2n_predict(
     if verbose:
         print("✓ Multi-output baseline system trained")
 
-    
     # --- Predict ---
     if verbose:
         print("Generating predictions...")
 
     predictions = predict_multivariate(
-        baseline_forecasters, steps_ahead=forecast_horizon
+        baseline_forecasters,
+        steps_ahead=forecast_horizon,
+        show_progress=show_progress,
     )
 
     return predictions