fixup! use pydantic.PositiveInt

openfun · Nov 29, 2023 · 9143d64 · 9143d64
1 parent c9c908e
commit 9143d64
Show file tree

Hide file tree

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