Skip to content

Commit ff1e21b

Browse files
committed
Fix Python 3.10 log capture import
1 parent 76215ea commit ff1e21b

1 file changed

Lines changed: 7 additions & 1 deletion

File tree

src/_pytask/logging.py

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -39,6 +39,12 @@
3939
pass
4040

4141

42+
if TYPE_CHECKING:
43+
LoggingStreamHandler = logging.StreamHandler[io.StringIO]
44+
else:
45+
LoggingStreamHandler = logging.StreamHandler
46+
47+
4248
DEFAULT_LOG_FORMAT = "%(levelname)-8s %(name)s:%(filename)s:%(lineno)d %(message)s"
4349
DEFAULT_LOG_DATE_FORMAT = "%H:%M:%S"
4450

@@ -287,7 +293,7 @@ def _humanize_time( # noqa: C901, PLR0912
287293
return result
288294

289295

290-
class LogCaptureHandler(logging.StreamHandler[io.StringIO]):
296+
class LogCaptureHandler(LoggingStreamHandler):
291297
"""Capture logs in a string buffer."""
292298

293299
def __init__(self) -> None:

0 commit comments

Comments
 (0)