Skip to content

Commit 00a1f04

Browse files
CarltonXiangharvey_xiang
andauthored
Feat/timer debug (#753)
* feat: timer false * feat: timer debug threshold * feat: timer debug threshold --------- Co-authored-by: harvey_xiang <[email protected]>
1 parent eb3c4f2 commit 00a1f04

File tree

1 file changed

+3
-5
lines changed

1 file changed

+3
-5
lines changed

src/memos/utils.py

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -6,9 +6,6 @@
66

77
logger = get_logger(__name__)
88

9-
# Global threshold (seconds) for timing logs
10-
DEFAULT_TIME_BAR = 10.0
11-
129

1310
def timed_with_status(
1411
func=None,
@@ -97,7 +94,7 @@ def wrapper(*args, **kwargs):
9794
return decorator(func)
9895

9996

100-
def timed(func=None, *, log=False, log_prefix=""):
97+
def timed(func=None, *, log=True, log_prefix=""):
10198
def decorator(fn):
10299
def wrapper(*args, **kwargs):
103100
start = time.perf_counter()
@@ -107,7 +104,8 @@ def wrapper(*args, **kwargs):
107104
if log is not True:
108105
return result
109106

110-
if elapsed_ms >= (DEFAULT_TIME_BAR * 1000.0):
107+
# 100ms threshold
108+
if elapsed_ms >= 100.0:
111109
logger.info(f"[TIMER] {log_prefix or fn.__name__} took {elapsed_ms:.0f} ms")
112110

113111
return result

0 commit comments

Comments
 (0)