Import FixtureRequest from pytest

datastax · Aug 20, 2024 · 10dbac5 · 10dbac5
1 parent eb54a9f
commit 10dbac5
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 7 deletions.
diff --git a/libs/colbert/tests/integration_tests/conftest.py b/libs/colbert/tests/integration_tests/conftest.py
@@ -1,7 +1,6 @@
 from typing import Iterator
 
 import pytest
-from _pytest.fixtures import FixtureRequest
 from cassandra.cluster import Session
 from ragstack_tests_utils import AstraDBTestStore, LocalCassandraTestStore
 
@@ -20,7 +19,7 @@ def astra_db() -> AstraDBTestStore:
 
 
 @pytest.fixture()
-def session(request: FixtureRequest) -> Session:
+def session(request: pytest.FixtureRequest) -> Session:
     test_store = request.getfixturevalue(request.param)
     session = test_store.create_cassandra_session()
     session.default_timeout = 180

diff --git a/libs/langchain/tests/integration_tests/conftest.py b/libs/langchain/tests/integration_tests/conftest.py
@@ -1,7 +1,6 @@
 from typing import Iterator
 
 import pytest
-from _pytest.fixtures import FixtureRequest
 from cassandra.cluster import Session
 from dotenv import load_dotenv
 from ragstack_tests_utils import AstraDBTestStore, LocalCassandraTestStore
@@ -22,13 +21,13 @@ def astra_db() -> AstraDBTestStore:
     return AstraDBTestStore()
 
 
-def get_session(request: FixtureRequest) -> Session:
+def get_session(request: pytest.FixtureRequest) -> Session:
     test_store = request.getfixturevalue(request.param)
     session = test_store.create_cassandra_session()
     session.default_timeout = 180
     return session
 
 
 @pytest.fixture()
-def session(request: FixtureRequest) -> Session:
+def session(request: pytest.FixtureRequest) -> Session:
     return get_session(request)
diff --git a/libs/llamaindex/tests/integration_tests/conftest.py b/libs/llamaindex/tests/integration_tests/conftest.py
@@ -1,7 +1,6 @@
 from typing import Iterator
 
 import pytest
-from _pytest.fixtures import FixtureRequest
 from cassandra.cluster import Session
 from ragstack_tests_utils import AstraDBTestStore, LocalCassandraTestStore
 
@@ -20,7 +19,7 @@ def astra_db() -> AstraDBTestStore:
 
 
 @pytest.fixture()
-def session(request: FixtureRequest) -> Session:
+def session(request: pytest.FixtureRequest) -> Session:
     test_store = request.getfixturevalue(request.param)
     session = test_store.create_cassandra_session()
     session.default_timeout = 180