diff --git a/tests/core/celery/tasks/test_collection_delete.py b/tests/core/celery/tasks/test_collection_delete.py index 27e6930fc2..581e532f2d 100644 --- a/tests/core/celery/tasks/test_collection_delete.py +++ b/tests/core/celery/tasks/test_collection_delete.py @@ -1,4 +1,4 @@ -from _pytest.logging import LogCaptureFixture +from pytest import LogCaptureFixture from sqlalchemy import select from sqlalchemy.orm import sessionmaker diff --git a/tests/core/celery/test_task.py b/tests/core/celery/test_task.py index edd2fe7bfc..ba785a7a7b 100644 --- a/tests/core/celery/test_task.py +++ b/tests/core/celery/test_task.py @@ -1,6 +1,6 @@ from unittest.mock import MagicMock, patch -from _pytest.logging import LogCaptureFixture +from pytest import LogCaptureFixture from sqlalchemy.pool import NullPool from core.celery.task import Task