diff --git a/scripts/train_lgbm_daily_high.py b/scripts/train_lgbm_daily_high.py index f0c7bfe1..b31788f1 100644 --- a/scripts/train_lgbm_daily_high.py +++ b/scripts/train_lgbm_daily_high.py @@ -4,14 +4,10 @@ import json import os import sys from datetime import datetime -from typing import TYPE_CHECKING, Any, Dict, List +from typing import Any, Dict, List ROOT_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__))) -if TYPE_CHECKING: - import lightgbm as lgb - import numpy as np - MODEL_PATH = os.path.join(ROOT_DIR, "artifacts", "models", "lgbm_daily_high.txt") SCHEMA_PATH = os.path.join(ROOT_DIR, "artifacts", "models", "lgbm_daily_high_schema.json")