#124: Corrected type hint for recipe param "_model" (#128)

* #124: Corrected type hint for recipe param "_model" * #124: Added changelog entry
model-bakers · Nov 23, 2020 · 69d110c · 69d110c
1 parent 244eae0
commit 69d110c
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -11,6 +11,8 @@ and this project adheres to [Semantic Versioning](http://semver.org/).
 
 ### Changed
 
+- Type hinting fixed for Recipe "_model" parameter
+
 ### Removed
 
 

diff --git a/model_bakery/recipe.py b/model_bakery/recipe.py
@@ -2,7 +2,6 @@
 from typing import Any, Dict, List, Type, Union, cast
 
 from django.db.models import Model
-from django.db.models.base import ModelBase
 
 from . import baker
 from .exceptions import RecipeNotFound
@@ -15,7 +14,7 @@
 
 
 class Recipe(object):
-    def __init__(self, _model: Union[str, Type[ModelBase]], **attrs) -> None:
+    def __init__(self, _model: Union[str, Type[Model]], **attrs) -> None:
         self.attr_mapping = attrs
         self._model = _model
         # _iterator_backups will hold values of the form (backup_iterator, usable_iterator).
Original file line number	Diff line number	Diff line change
Expand Up		@@ -11,6 +11,8 @@ and this project adheres to [Semantic Versioning](http://semver.org/).

		### Changed

		- Type hinting fixed for Recipe "_model" parameter

		### Removed


Expand Down