Skip to content

Commit 930c25c

Browse files
committed
Replace loguru with standard logging module
1 parent 2758756 commit 930c25c

5 files changed

Lines changed: 22 additions & 12 deletions

File tree

src/fyp/selfplay/proposer.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,18 +6,20 @@
66
"""
77

88
import json
9+
import logging
910
import random
1011
from dataclasses import dataclass, field
1112
from datetime import datetime
1213

1314
import numpy as np
14-
from loguru import logger
1515

1616
from fyp.selfplay.utils import (
1717
apply_scenario_transformation,
1818
estimate_scenario_difficulty,
1919
)
2020

21+
logger = logging.getLogger(__name__)
22+
2123

2224
@dataclass
2325
class ScenarioProposal:

src/fyp/selfplay/solver.py

Lines changed: 10 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -4,18 +4,11 @@
44
data and proposed scenarios, using the PatchTST architecture with quantile regression.
55
"""
66

7+
import logging
78
import os
89
from typing import Any
910

1011
import 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

2013
from fyp.data_loader import EnergyDataLoader
2114
from fyp.selfplay.proposer import ScenarioProposal
@@ -24,6 +17,15 @@
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
2830
try:
2931
from fyp.models.patchtst import PatchTSTForecaster, create_patchtst_config

src/fyp/selfplay/trainer.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,13 +6,13 @@
66
"""
77

88
import json
9+
import logging
910
import os
1011
import time
1112
from datetime import datetime
1213
from typing import Any
1314

1415
import numpy as np
15-
from loguru import logger
1616
from tqdm import tqdm
1717

1818
from fyp.data_loader import EnergyDataLoader
@@ -22,6 +22,8 @@
2222
from fyp.selfplay.utils import create_sliding_windows
2323
from fyp.selfplay.verifier import VerifierAgent
2424

25+
logger = logging.getLogger(__name__)
26+
2527

2628
class SelfPlayTrainer:
2729
"""Orchestrates propose→solve→verify self-play training loop."""

src/fyp/selfplay/utils.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,11 @@
11
"""Utility functions for self-play training system."""
22

3+
import logging
34

45
import numpy as np
56
import pandas as pd
6-
from loguru import logger
7+
8+
logger = logging.getLogger(__name__)
79

810

911
def create_sliding_windows(

src/fyp/selfplay/verifier.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,11 +6,13 @@
66
"""
77

88
import json
9+
import logging
910
from abc import ABC, abstractmethod
1011
from typing import TYPE_CHECKING, Optional
1112

1213
import numpy as np
13-
from loguru import logger
14+
15+
logger = logging.getLogger(__name__)
1416

1517
if TYPE_CHECKING:
1618
from fyp.selfplay.proposer import ScenarioProposal

0 commit comments

Comments
 (0)