From 10df3f31f4b066a83302b01b7199da048abafdb0 Mon Sep 17 00:00:00 2001 From: Cody Fincher Date: Sun, 15 Dec 2024 18:06:34 +0000 Subject: [PATCH] fix: patch picologging tests --- tests/helpers.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/tests/helpers.py b/tests/helpers.py index b541ed28ec..97897b8e88 100644 --- a/tests/helpers.py +++ b/tests/helpers.py @@ -8,7 +8,11 @@ from contextlib import AbstractContextManager, contextmanager from typing import Any, AsyncContextManager, Awaitable, ContextManager, Generator, TypeVar, cast, overload -import picologging +try: + import picologging +except ImportError: + import logging as picologging + from _pytest.logging import LogCaptureHandler, _LiveLoggingNullHandler from litestar._openapi.schema_generation import SchemaCreator