spotforecast2 0.1.1__tar.gz → 0.2.0__tar.gz

{spotforecast2-0.1.1 → spotforecast2-0.2.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: spotforecast2
-Version: 0.1.1
+Version: 0.2.0
 Summary: Forecasting with spot
 Author: bartzbeielstein
 Author-email: bartzbeielstein <32470350+bartzbeielstein@users.noreply.github.com>

{spotforecast2-0.1.1 → spotforecast2-0.2.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "spotforecast2"
-version = "0.1.1"
+version = "0.2.0"
 description = "Forecasting with spot"
 readme = "README.md"
 authors = [

{spotforecast2-0.1.1 → spotforecast2-0.2.0}/src/spotforecast2/processing/n2n_predict.py

@@ -210,6 +210,7 @@ def _model_directory_exists(model_dir: Union[str, Path]) -> bool:
 
 
 def n2n_predict(
+    data: Optional[pd.DataFrame] = None,
     columns: Optional[List[str]] = None,
     forecast_horizon: int = 24,
     contamination: float = 0.01,
@@ -233,6 +234,8 @@ def n2n_predict(
     Existing models are reused for prediction unless force_train=True.
 
     Args:
+        data: Optional DataFrame with target time series data. If None, fetches data automatically.
+            Default: None.
         columns: List of target columns to forecast. If None, uses all available columns.
             Default: None.
         forecast_horizon: Number of time steps to forecast ahead. Default: 24.
@@ -314,7 +317,11 @@ def n2n_predict(
         print("Fetching data...")
 
     # Fetch data
-    data = fetch_data(columns=TARGET)
+    if data is not None:
+        if TARGET is not None:
+            data = data[TARGET]
+    else:
+        data = fetch_data(columns=TARGET)
 
     START, END, COV_START, COV_END = get_start_end(
         data=data,

{spotforecast2-0.1.1 → spotforecast2-0.2.0}/src/spotforecast2/processing/n2n_predict_with_covariates.py

@@ -727,6 +727,7 @@ def _model_directory_exists(model_dir: Union[str, Path]) -> bool:
 
 
 def n2n_predict_with_covariates(
+    data: Optional[pd.DataFrame] = None,
     forecast_horizon: int = 24,
     contamination: float = 0.01,
     window_size: int = 72,
@@ -763,6 +764,8 @@ def n2n_predict_with_covariates(
     Existing models are reused for prediction unless force_train=True.
 
     Args:
+        data: Optional DataFrame with target time series data. If None, fetches data automatically.
+            Default: None.
         forecast_horizon: Number of time steps to forecast ahead. Default: 24.
         contamination: Contamination parameter for outlier detection. Default: 0.01.
         window_size: Rolling window size for gap detection. Default: 72.
@@ -865,7 +868,8 @@ def n2n_predict_with_covariates(
     if verbose:
         print("\n[1/9] Loading and preparing target data...")
 
-    data = fetch_data()
+    if data is None:
+        data = fetch_data()
     target_columns = data.columns.tolist()
 
     if verbose: