From fb42aa861cbbbadec49955c9f973015a5a9e16e3 Mon Sep 17 00:00:00 2001 From: Serhii Tereshchenko Date: Wed, 17 Jan 2024 13:05:22 +0200 Subject: [PATCH] Apply suggestions from code review Co-authored-by: Viicos --- factory/base.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/factory/base.py b/factory/base.py index 84c151fe..7009efa1 100644 --- a/factory/base.py +++ b/factory/base.py @@ -514,7 +514,7 @@ def build(cls, **kwargs) -> T: return cls._generate(enums.BUILD_STRATEGY, kwargs) @classmethod - def build_batch(cls, size, **kwargs) -> List[T]: + def build_batch(cls, size: int, **kwargs) -> List[T]: """Build a batch of instances of the given class, with overridden attrs. Args: @@ -531,7 +531,7 @@ def create(cls, **kwargs) -> T: return cls._generate(enums.CREATE_STRATEGY, kwargs) @classmethod - def create_batch(cls, size, **kwargs) -> List[T]: + def create_batch(cls, size: int, **kwargs) -> List[T]: """Create a batch of instances of the given class, with overridden attrs. Args: