|
10 | 10 | import sys |
11 | 11 | from datetime import datetime, timedelta |
12 | 12 | from pathlib import Path |
13 | | -from typing import Any, Optional |
| 13 | +from typing import Any |
14 | 14 |
|
15 | 15 | import pandas as pd |
16 | 16 | from prefect import flow, task |
@@ -79,7 +79,7 @@ def backup_current_model( |
79 | 79 | @task(name="prepare-retraining-data") |
80 | 80 | def prepare_retraining_data( |
81 | 81 | original_data_path: str, |
82 | | - new_data_path: Optional[str] = None, |
| 82 | + new_data_path: str | None = None, |
83 | 83 | data_window_days: int = 365, |
84 | 84 | ) -> tuple[pd.DataFrame, pd.DataFrame, dict]: |
85 | 85 | """Prepare data for retraining with new observations.""" |
@@ -159,7 +159,7 @@ def train_new_model( |
159 | 159 | train_data: pd.DataFrame, |
160 | 160 | val_data: pd.DataFrame, |
161 | 161 | model_type: str = "random_forest", |
162 | | - hyperparameters: Optional[dict] = None, |
| 162 | + hyperparameters: dict | None = None, |
163 | 163 | ) -> tuple[str, dict]: |
164 | 164 | """Train a new model with the prepared data.""" |
165 | 165 | logger = get_run_logger() |
@@ -324,7 +324,7 @@ def deploy_new_model( |
324 | 324 | model_file_path: str, |
325 | 325 | model_path: str, |
326 | 326 | validation_results: dict, |
327 | | - deployment_metadata: Optional[dict] = None, |
| 327 | + deployment_metadata: dict | None = None, |
328 | 328 | ) -> dict: |
329 | 329 | """Deploy the new model if validation passed.""" |
330 | 330 | logger = get_run_logger() |
@@ -467,12 +467,12 @@ def automated_retraining_flow( |
467 | 467 | triggers: list[str], |
468 | 468 | model_path: str = "models/model.pkl", |
469 | 469 | training_data_path: str = "data/real_data/premier_league_matches.parquet", |
470 | | - new_data_path: Optional[str] = None, |
| 470 | + new_data_path: str | None = None, |
471 | 471 | backup_dir: str = "models/backups", |
472 | 472 | model_type: str = "random_forest", |
473 | 473 | min_accuracy_threshold: float = 0.45, |
474 | 474 | improvement_threshold: float = 0.01, |
475 | | - hyperparameters: Optional[dict] = None, |
| 475 | + hyperparameters: dict | None = None, |
476 | 476 | ) -> dict[str, Any]: |
477 | 477 | """ |
478 | 478 | Main automated retraining flow. |
|
0 commit comments