From 9143d64e739b5fe86ed155d9be3dea5b8b8dcd99 Mon Sep 17 00:00:00 2001 From: SergioSim Date: Wed, 29 Nov 2023 13:48:42 +0100 Subject: [PATCH] fixup! use pydantic.PositiveInt --- src/ralph/backends/data/async_es.py | 2 +- src/ralph/backends/data/async_mongo.py | 2 +- src/ralph/backends/data/base.py | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/src/ralph/backends/data/async_es.py b/src/ralph/backends/data/async_es.py index 0670cd88c..311ddf137 100644 --- a/src/ralph/backends/data/async_es.py +++ b/src/ralph/backends/data/async_es.py @@ -217,7 +217,7 @@ async def write( # noqa: PLR0913 chunk_size: Optional[int] = None, ignore_errors: bool = False, operation_type: Optional[BaseOperationType] = None, - concurrency: Optional[int] = None, + concurrency: Optional[PositiveInt] = None, ) -> int: """Write data documents to the target index and return their count. diff --git a/src/ralph/backends/data/async_mongo.py b/src/ralph/backends/data/async_mongo.py index 77f1183c2..811e10388 100644 --- a/src/ralph/backends/data/async_mongo.py +++ b/src/ralph/backends/data/async_mongo.py @@ -196,7 +196,7 @@ async def write( # noqa: PLR0913 chunk_size: Optional[int] = None, ignore_errors: bool = False, operation_type: Optional[BaseOperationType] = None, - concurrency: Optional[int] = None, + concurrency: Optional[PositiveInt] = None, ) -> int: """Write data documents to the target collection and return their count. diff --git a/src/ralph/backends/data/base.py b/src/ralph/backends/data/base.py index 772506a48..308a1423b 100644 --- a/src/ralph/backends/data/base.py +++ b/src/ralph/backends/data/base.py @@ -357,7 +357,7 @@ async def write( # noqa: PLR0913 chunk_size: Optional[int] = None, ignore_errors: bool = False, operation_type: Optional[BaseOperationType] = None, - concurrency: Optional[int] = None, + concurrency: Optional[PositiveInt] = None, ) -> int: """Write `data` records to the `target` container and return their count.