File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change 66"""
77
88import json
9+ import logging
910import random
1011from dataclasses import dataclass , field
1112from datetime import datetime
1213
1314import numpy as np
14- from loguru import logger
1515
1616from fyp .selfplay .utils import (
1717 apply_scenario_transformation ,
1818 estimate_scenario_difficulty ,
1919)
2020
21+ logger = logging .getLogger (__name__ )
22+
2123
2224@dataclass
2325class ScenarioProposal :
Original file line number Diff line number Diff line change 44data and proposed scenarios, using the PatchTST architecture with quantile regression.
55"""
66
7+ import logging
78import os
89from typing import Any
910
1011import numpy as np
11- from loguru import logger
12-
13- # Lazy import torch to avoid dependency issues during testing
14- try :
15- import torch
16- except ImportError :
17- torch = None
18- logger .warning ("PyTorch not available. Some solver functionality will be limited." )
1912
2013from fyp .data_loader import EnergyDataLoader
2114from fyp .selfplay .proposer import ScenarioProposal
2417 create_sliding_windows ,
2518)
2619
20+ logger = logging .getLogger (__name__ )
21+
22+ # Lazy import torch to avoid dependency issues during testing
23+ try :
24+ import torch
25+ except ImportError :
26+ torch = None
27+ logger .warning ("PyTorch not available. Some solver functionality will be limited." )
28+
2729# Lazy import PatchTST to avoid torch dependency
2830try :
2931 from fyp .models .patchtst import PatchTSTForecaster , create_patchtst_config
Original file line number Diff line number Diff line change 66"""
77
88import json
9+ import logging
910import os
1011import time
1112from datetime import datetime
1213from typing import Any
1314
1415import numpy as np
15- from loguru import logger
1616from tqdm import tqdm
1717
1818from fyp .data_loader import EnergyDataLoader
2222from fyp .selfplay .utils import create_sliding_windows
2323from fyp .selfplay .verifier import VerifierAgent
2424
25+ logger = logging .getLogger (__name__ )
26+
2527
2628class SelfPlayTrainer :
2729 """Orchestrates propose→solve→verify self-play training loop."""
Original file line number Diff line number Diff line change 11"""Utility functions for self-play training system."""
22
3+ import logging
34
45import numpy as np
56import pandas as pd
6- from loguru import logger
7+
8+ logger = logging .getLogger (__name__ )
79
810
911def create_sliding_windows (
Original file line number Diff line number Diff line change 66"""
77
88import json
9+ import logging
910from abc import ABC , abstractmethod
1011from typing import TYPE_CHECKING , Optional
1112
1213import numpy as np
13- from loguru import logger
14+
15+ logger = logging .getLogger (__name__ )
1416
1517if TYPE_CHECKING :
1618 from fyp .selfplay .proposer import ScenarioProposal
You can’t perform that action at this time.
0 commit comments