Fixed timeit issue. (#531)

daxa-ai · Sep 5, 2024 · ffbc14b · ffbc14b
1 parent 8a884b6
commit ffbc14b
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/pebblo/app/utils/utils.py b/pebblo/app/utils/utils.py
@@ -297,6 +297,7 @@ def get_current_time():
 def timeit(func):
     def wrapper(*args, **kwargs):
         if logger.isEnabledFor(logging.DEBUG):
+            # Apply time profiling
             start_time = time.perf_counter()
             response = func(*args, **kwargs)
             end_time = time.perf_counter()
@@ -305,6 +306,8 @@ def wrapper(*args, **kwargs):
             )
             return response
         else:
-            return {}
+            # Skip time profiling
+            response = func(*args, **kwargs)
+            return response
 
     return wrapper