From 2f5b47c8b30d1b7d86443391332be2f3805cdafd Mon Sep 17 00:00:00 2001 From: Scott Gasch Date: Tue, 8 Feb 2022 20:21:02 -0800 Subject: Cleanup more contextlib.AbstractContextManagers and Literal[False]s. --- stopwatch.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) (limited to 'stopwatch.py') diff --git a/stopwatch.py b/stopwatch.py index fa4f2b5..ae8e01a 100644 --- a/stopwatch.py +++ b/stopwatch.py @@ -2,11 +2,12 @@ """A simple stopwatch decorator / context for timing things.""" +import contextlib import time -from typing import Callable, Optional +from typing import Callable, Literal -class Timer(object): +class Timer(contextlib.AbstractContextManager): """ A stopwatch to time how long something takes (walltime). @@ -31,6 +32,6 @@ class Timer(object): self.end = 0.0 return lambda: self.end - self.start - def __exit__(self, *args) -> Optional[bool]: + def __exit__(self, *args) -> Literal[False]: self.end = time.perf_counter() - return None # don't suppress exceptions + return False -- cgit v1.3