We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent b5964fd commit c3be9a4Copy full SHA for c3be9a4
src/llmcompressor/metrics/logger.py
@@ -389,11 +389,13 @@ def _log_lambda(
389
def is_higher_than_debug(lev: int | str | None = None) -> bool:
390
"""Check if the given level is higher than DEBUG level."""
391
debug_level_no = logger.level("DEBUG").no
392
- if isinstance(lev, int):
393
- return level > debug_level_no
394
- elif isinstance(lev, str):
395
- return logger.level(lev).no > debug_level_no
396
- return False
+ match lev:
+ case int():
+ return lev > debug_level_no
+ case str():
+ return logger.level(lev).no > debug_level_no
397
+ case _:
398
+ return False
399
400
if is_higher_than_debug(level):
401
if step is not None:
0 commit comments