Skip to content

make it less noisy + style #1

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
eyalk11 wants to merge 1 commit into
base: main
Choose a base branch
from
Open

make it less noisy + style #1

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
23 changes: 13 additions & 10 deletions execution_timer/execution_timer.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -256,7 +256,8 @@ def __init__(
"""

save_measure = CHECK_TYPE(save_measure, "save_measure", bool, True)
return_measure = CHECK_TYPE(return_measure, "return_measure", bool, False)
return_measure = CHECK_TYPE(
return_measure, "return_measure", bool, False)
nanoseconds = CHECK_TYPE(nanoseconds, "nanoseconds", bool, False)
n_iter = CHECK_TYPE(n_iter, "n_iter", int, 1)
n_iter = (
Expand All @@ -281,6 +282,7 @@ def time_execution(
func: Callable[..., Any] = None,
*,
return_measure: Optional[bool] = None,
noisy: Optional[bool] = None,
) -> Union[Callable[..., Any], Any]:
"""
Decorator method to measure the execution time of a function or method.
Expand Down Expand Up @@ -329,21 +331,24 @@ def wrapper(*args: tuple, **kwargs: dict) -> Union[Any, tuple[Any, float]]:
if class_name not in self.execution_times:
self.execution_times[class_name] = {}
if func_name not in self.execution_times[class_name]:
self.execution_times[class_name][func_name] = []
self.execution_times[class_name][func_name] = [
]
self.execution_times[class_name][func_name].append(
elapsed_time
)
else:
if func_name not in self.execution_times:
self.execution_times[func_name] = []
self.execution_times[func_name].append(elapsed_time)
self.execution_times[func_name].append(
elapsed_time)

if return_measure:
return result, elapsed_time
else:
print(
f"The '{inner_func.__name__}' function was executed in {elapsed_time:.5f} {'nanoseconds' if self.nanoseconds else 'seconds'}."
)
if noisy:
print(
f"The '{inner_func.__name__}' function was executed in {elapsed_time:.5f} {'nanoseconds' if self.nanoseconds else 'seconds'}."
)
return result

return wrapper
Expand Down Expand Up @@ -389,9 +394,7 @@ def average_measured_time(self) -> dict[str, Optional[float]]:
elif isinstance(value, dict):
class_averages: dict[str, float] = {}
for func_name, times in value.items():
class_averages[func_name] = (
sum(times) / len(times) if times else None
)
averages[key] = class_averages
class_averages[func_name] = {"avg": sum(
times) / len(times), "times": len(times), "cumulative": sum(times)} if times else {}

return averages