diff options
Diffstat (limited to 'factory/mongoengine.py')
-rw-r--r-- | factory/mongoengine.py | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/factory/mongoengine.py b/factory/mongoengine.py index 729ebb1..e3ab99c 100644 --- a/factory/mongoengine.py +++ b/factory/mongoengine.py @@ -37,12 +37,12 @@ class MongoEngineFactory(base.Factory): abstract = True @classmethod - def _build(cls, target_class, *args, **kwargs): - return target_class(*args, **kwargs) + def _build(cls, model_class, *args, **kwargs): + return model_class(*args, **kwargs) @classmethod - def _create(cls, target_class, *args, **kwargs): - instance = target_class(*args, **kwargs) + def _create(cls, model_class, *args, **kwargs): + instance = model_class(*args, **kwargs) if instance._is_document: instance.save() return instance |