From 5bfcfa8d958366811a3869b0bf3372593bed075d Mon Sep 17 00:00:00 2001 From: J08nY Date: Wed, 7 Dec 2022 13:52:05 +0100 Subject: Fix perf issue with pyinstrument. --- test/utils.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) (limited to 'test/utils.py') diff --git a/test/utils.py b/test/utils.py index a74fae4..d4893eb 100644 --- a/test/utils.py +++ b/test/utils.py @@ -66,10 +66,10 @@ class Profiler: else: self._prof.print_stats("cumtime") - def get_time(self): + def get_time(self) -> float: if self._state != "out": raise ValueError if self._prof_type == "py": - return self._root_frame.time() + return self._root_frame.time else: - return pstats.Stats(self._prof).total_tt + return pstats.Stats(self._prof).total_tt # type: ignore -- cgit v1.2.3-70-g09d2