Allow functions as models

FactoryBoy · Apr 9, 2024 · c894d2c · c894d2c
1 parent dca3a16
commit c894d2c
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/factory/base.py b/factory/base.py
@@ -1,6 +1,7 @@
 # Copyright: See the LICENSE file.
 
 
+import inspect
 import collections
 import logging
 import warnings
@@ -243,6 +244,7 @@ def _get_counter_reference(self):
         if (self.model is not None
                 and self.base_factory is not None
                 and self.base_factory._meta.model is not None
+                and inspect.isclass(self.model)
                 and issubclass(self.model, self.base_factory._meta.model)):
             return self.base_factory._meta.counter_reference
         else: