diff options
Diffstat (limited to 'test/utils.py')
| -rw-r--r-- | test/utils.py | 6 |
1 files changed, 3 insertions, 3 deletions
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 |