Add py.typed files (#618)

Author: cbornet

libs/colbert/ragstack_colbert/py.typed

@@ -1,11 +1,13 @@
+from typing import Iterator
+
 import pytest
 from _pytest.fixtures import FixtureRequest
 from cassandra.cluster import Session
 from ragstack_tests_utils import AstraDBTestStore, LocalCassandraTestStore
 
 
 @pytest.fixture(scope="session")
-def cassandra() -> LocalCassandraTestStore:
+def cassandra() -> Iterator[LocalCassandraTestStore]:
     store = LocalCassandraTestStore()
     yield store
     if store.docker_container:

libs/colbert/tests/integration_tests/conftest.py

@@ -1,3 +1,5 @@
+from typing import Iterator
+
 import pytest
 from cassandra.cluster import Session
 from dotenv import load_dotenv
@@ -7,7 +9,7 @@
 
 
 @pytest.fixture(scope="session")
-def cassandra() -> LocalCassandraTestStore:
+def cassandra() -> Iterator[LocalCassandraTestStore]:
     store = LocalCassandraTestStore()
     yield store
     if store.docker_container:

libs/knowledge-graph/ragstack_knowledge_graph/py.typed

@@ -1,11 +1,13 @@
+from typing import Iterator
+
 import pytest
 from _pytest.fixtures import FixtureRequest
 from cassandra.cluster import Session
 from ragstack_tests_utils import AstraDBTestStore, LocalCassandraTestStore
 
 
 @pytest.fixture(scope="session")
-def cassandra() -> LocalCassandraTestStore:
+def cassandra() -> Iterator[LocalCassandraTestStore]:
     store = LocalCassandraTestStore()
     yield store
     if store.docker_container: