Skip to content

Commit fb42aa8

Browse files
last-partizanViicos
andcommitted
Apply suggestions from code review
Co-authored-by: Viicos <victo.plot@gmail.com>
1 parent 1658cb2 commit fb42aa8

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

factory/base.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -514,7 +514,7 @@ def build(cls, **kwargs) -> T:
514514
return cls._generate(enums.BUILD_STRATEGY, kwargs)
515515

516516
@classmethod
517-
def build_batch(cls, size, **kwargs) -> List[T]:
517+
def build_batch(cls, size: int, **kwargs) -> List[T]:
518518
"""Build a batch of instances of the given class, with overridden attrs.
519519
520520
Args:
@@ -531,7 +531,7 @@ def create(cls, **kwargs) -> T:
531531
return cls._generate(enums.CREATE_STRATEGY, kwargs)
532532

533533
@classmethod
534-
def create_batch(cls, size, **kwargs) -> List[T]:
534+
def create_batch(cls, size: int, **kwargs) -> List[T]:
535535
"""Create a batch of instances of the given class, with overridden attrs.
536536
537537
Args:

0 commit comments

Comments
 (0)