Skip to content

Commit 09a6e24

Browse files
authored
Set logger stacklevel (flagos-ai#1113)
### PR Category <!-- One of [ Train | Inference | Compress | Serve | RL | Core | Hardware | CICD | Tools | Others ] --> Others ### PR Types <!-- One of [ User Experience | New Features | Bug Fixes | Improvements | Performance | Breaking Change| Deprecations | Test Case | Docs | Others ] --> Improvements ### PR Description <!-- Describe what you’ve done --> Fix the logger stacklevel to point to the real location of the function e.g. before: [2026-02-09 19:51:00,204 FlagScale logger.py:25 INFO] after: [2026-02-10 11:58:06,728 FlagScale inference_qwen_gr00t.py:50 INFO]
1 parent da4dc54 commit 09a6e24

1 file changed

Lines changed: 5 additions & 5 deletions

File tree

flagscale/logger.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -22,19 +22,19 @@ def __init__(self, name, level=logging.INFO):
2222
self.logger.addHandler(stream_handler)
2323

2424
def info(self, message):
25-
self.logger.info(message)
25+
self.logger.info(message, stacklevel=2)
2626

2727
def warning(self, message):
28-
self.logger.warning(message)
28+
self.logger.warning(message, stacklevel=2)
2929

3030
def error(self, message):
31-
self.logger.error(message)
31+
self.logger.error(message, stacklevel=2)
3232

3333
def critical(self, message):
34-
self.logger.critical(message)
34+
self.logger.critical(message, stacklevel=2)
3535

3636
def debug(self, message):
37-
self.logger.debug(message)
37+
self.logger.debug(message, stacklevel=2)
3838

3939

4040
GLOBAL_LOGGER = None

0 commit comments

Comments
 (0)