diff --git a/django_fakery/types.py b/django_fakery/types.py index d553c32..983af82 100644 --- a/django_fakery/types.py +++ b/django_fakery/types.py @@ -7,6 +7,6 @@ FieldMap = Dict[str, Any] Lookup = Dict[str, Any] SaveHooks = List[Callable[[T], None]] -LazySaveHooks = List[Callable[[models.Model], None]] +LazySaveHooks = List[Callable[[models.Model], Any]] Built = Tuple[T, Dict[str, List[models.Model]]] LazyBuilt = Tuple[models.Model, Dict[str, List[models.Model]]]