Clean up pytest imports

ThePalaceProject · Apr 11, 2024 · b2dda94 · b2dda94
1 parent 14096aa
commit b2dda94
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git 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
@@ -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