nucliadb-dataset 6.9.1.post5187__py3-none-any.whl → 6.10.0.post5647__py3-none-any.whl

nucliadb_dataset/tests/fixtures.py

@@ -14,7 +14,7 @@
 
 import re
 import tempfile
-from typing import AsyncIterator, Iterator, Optional
+from typing import TYPE_CHECKING, AsyncIterator, Iterator, Optional
 
 import docker  # type: ignore
 import grpc
@@ -22,7 +22,6 @@ import pytest
 from grpc import aio
 
 from nucliadb_models.common import UserClassification
-from nucliadb_models.entities import CreateEntitiesGroupPayload, Entity
 from nucliadb_models.labels import Label, LabelSet, LabelSetKind
 from nucliadb_models.metadata import UserMetadata
 from nucliadb_models.resource import KnowledgeBoxObj
@@ -35,6 +34,9 @@ from nucliadb_sdk.v2.sdk import NucliaDB
 DOCKER_ENV_GROUPS = re.search(r"//([^:]+)", docker.from_env().api.base_url)
 DOCKER_HOST: Optional[str] = DOCKER_ENV_GROUPS.group(1) if DOCKER_ENV_GROUPS else None
 
+if TYPE_CHECKING:
+    from nucliadb_protos.writer_pb2_grpc import WriterAsyncStub
+
 
 @pytest.fixture(scope="function")
 def upload_data_field_classification(sdk: NucliaDB, kb: KnowledgeBoxObj):
@@ -156,70 +158,6 @@ def upload_data_paragraph_classification(sdk: NucliaDB, kb: KnowledgeBoxObj):
     return kb
 
 
-@pytest.fixture(scope="function")
-def upload_data_token_classification(sdk: NucliaDB, kb: KnowledgeBoxObj):
-    sdk.create_entitygroup(
-        kbid=kb.uuid,
-        content=CreateEntitiesGroupPayload(
-            group="PERSON",
-            entities={
-                "ramon": Entity(value="Ramon"),
-                "carmen": Entity(value="Carmen Iniesta"),
-                "eudald": Entity(value="Eudald Camprubi"),
-            },
-            title="Animals",
-            color="black",
-        ),
-    )
-
-    sdk.create_entitygroup(
-        kbid=kb.uuid,
-        content=CreateEntitiesGroupPayload(
-            group="ANIMAL",
-            entities={
-                "cheetah": Entity(value="cheetah"),
-                "tiger": Entity(value="tiger"),
-                "lion": Entity(value="lion"),
-            },
-            title="Animals",
-            color="black",
-        ),
-    )
-    sdk.create_resource(
-        kbid=kb.uuid,
-        content=CreateResourcePayload(
-            slug=SlugString("doc1"),
-            texts={FieldIdString("text"): TextField(body="Ramon This is my lovely text")},
-        ),
-    )
-
-    sdk.create_resource(
-        kbid=kb.uuid,
-        content=CreateResourcePayload(
-            slug=SlugString("doc2"),
-            texts={
-                FieldIdString("text"): TextField(
-                    body="Carmen Iniesta shows an amazing classifier to Eudald Camprubi"
-                )
-            },
-        ),
-    )
-
-    sdk.create_resource(
-        kbid=kb.uuid,
-        content=CreateResourcePayload(
-            slug=SlugString("doc3"),
-            texts={
-                FieldIdString("text"): TextField(
-                    body="Which is the fastest animal, a lion, a tiger or a cheetah?"
-                )
-            },
-        ),
-    )
-
-    return kb
-
-
 @pytest.fixture(scope="function")
 def text_editors_kb(sdk: NucliaDB, kb: KnowledgeBoxObj):
     sdk.create_resource(
@@ -285,10 +223,10 @@ def temp_folder():
 
 
 @pytest.fixture
-async def ingest_stub(nucliadb) -> AsyncIterator[WriterStub]:
+async def ingest_stub(nucliadb) -> AsyncIterator["WriterAsyncStub"]:
     channel = aio.insecure_channel(f"{nucliadb.host}:{nucliadb.grpc}")
     stub = WriterStub(channel)
-    yield stub
+    yield stub  # type: ignore
     await channel.close(grace=True)
 
 
@@ -296,5 +234,5 @@ async def ingest_stub(nucliadb) -> AsyncIterator[WriterStub]:
 def ingest_stub_sync(nucliadb) -> Iterator[WriterStub]:
     channel = grpc.insecure_channel(f"{nucliadb.host}:{nucliadb.grpc}")
     stub = WriterStub(channel)
-    yield stub
+    yield stub  # type: ignore
     channel.close()

{nucliadb_dataset-6.9.1.post5187.dist-info → nucliadb_dataset-6.10.0.post5647.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nucliadb_dataset
-Version: 6.9.1.post5187
+Version: 6.10.0.post5647
 Summary: NucliaDB Train Python client
 Author-email: Nuclia <nucliadb@nuclia.com>
 License-Expression: Apache-2.0
@@ -10,12 +10,11 @@ Classifier: Development Status :: 3 - Alpha
 Classifier: Intended Audience :: Developers
 Classifier: Intended Audience :: Information Technology
 Classifier: Programming Language :: Python
-Classifier: Programming Language :: Python :: 3.9
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3 :: Only
-Requires-Python: <4,>=3.9
+Requires-Python: <4,>=3.10
 Description-Content-Type: text/markdown
 Requires-Dist: protobuf
 Requires-Dist: types-protobuf
@@ -25,9 +24,9 @@ Requires-Dist: aiohttp
 Requires-Dist: argdantic
 Requires-Dist: pydantic-settings>=2.2
 Requires-Dist: pyarrow
-Requires-Dist: nucliadb-protos>=6.9.1.post5187
-Requires-Dist: nucliadb-sdk>=6.9.1.post5187
-Requires-Dist: nucliadb-models>=6.9.1.post5187
+Requires-Dist: nucliadb-protos>=6.10.0.post5647
+Requires-Dist: nucliadb-sdk>=6.10.0.post5647
+Requires-Dist: nucliadb-models>=6.10.0.post5647
 
 # NUCLIADB TRAIN CLIENT
 

{nucliadb_dataset-6.9.1.post5187.dist-info → nucliadb_dataset-6.10.0.post5647.dist-info}/RECORD

@@ -10,9 +10,9 @@ nucliadb_dataset/settings.py,sha256=9NYeJVgIHbLqCenUFRwj6Iz9S7klOROXQfzaUiBUEl0,
 nucliadb_dataset/streamer.py,sha256=aBzYWNQQVWM5qA14f8p0YQdshzC4KpOQFt3J0TYe1uk,3060
 nucliadb_dataset/tasks.py,sha256=198o37vDlzS7OdXrHYhtwI8kz2WHWJnxpholh-rtTPQ,6227
 nucliadb_dataset/tests/__init__.py,sha256=zG33bUz1rHFPtvqQPWn4rDwBJt3FJodGuQYD45quiQg,583
-nucliadb_dataset/tests/fixtures.py,sha256=Fx82tfwoFx0YyaiC6tg-ThUDvS8pcmFSE0Pm1KpAS0I,9242
-nucliadb_dataset-6.9.1.post5187.dist-info/METADATA,sha256=8TRTk350GTFzteUvygWVPGNjmay4cBY-dhjoed2vVMY,1263
-nucliadb_dataset-6.9.1.post5187.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-nucliadb_dataset-6.9.1.post5187.dist-info/entry_points.txt,sha256=ORrWnn6AUFfHGY1fWPRPTz99KV_pXXwttZAopyT8qvQ,60
-nucliadb_dataset-6.9.1.post5187.dist-info/top_level.txt,sha256=aJtDe54tz6060E0uyk1rdTRAU4FPWo5if1fYFQGvdqU,17
-nucliadb_dataset-6.9.1.post5187.dist-info/RECORD,,
+nucliadb_dataset/tests/fixtures.py,sha256=5Ac20TKYmGe61d74yRT6hj0ENdLr3jT2Gbj2fiSX9LM,7510
+nucliadb_dataset-6.10.0.post5647.dist-info/METADATA,sha256=bzo2_22-tenSkpguiIJFXWls9b6DfHNCEEIZG9eHxIU,1218
+nucliadb_dataset-6.10.0.post5647.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+nucliadb_dataset-6.10.0.post5647.dist-info/entry_points.txt,sha256=ORrWnn6AUFfHGY1fWPRPTz99KV_pXXwttZAopyT8qvQ,60
+nucliadb_dataset-6.10.0.post5647.dist-info/top_level.txt,sha256=aJtDe54tz6060E0uyk1rdTRAU4FPWo5if1fYFQGvdqU,17
+nucliadb_dataset-6.10.0.post5647.dist-info/RECORD,,