Skip to content

Commit

Permalink
model-bakers#124: Corrected type hint for recipe param "_model"
Browse files Browse the repository at this point in the history
  • Loading branch information
Ronny Vedrilla committed Nov 13, 2020
1 parent 244eae0 commit 17613b2
Showing 1 changed file with 1 addition and 2 deletions.
3 changes: 1 addition & 2 deletions model_bakery/recipe.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand All @@ -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).
Expand Down

0 comments on commit 17613b2

Please sign in to comment.