diff --git a/polyfactory/factories/base.py b/polyfactory/factories/base.py index e109d7fd..c6c96c08 100644 --- a/polyfactory/factories/base.py +++ b/polyfactory/factories/base.py @@ -239,10 +239,7 @@ class Foo(ModelFactory[MyModel]): # <<< MyModel generic_args: Sequence[type[T]] = [ arg for factory_base in factory_bases for arg in get_args(factory_base) if not isinstance(arg, TypeVar) ] - if len(generic_args) != 1: - return None - - return generic_args[0] + return None if len(generic_args) != 1 else generic_args[0] @classmethod def _get_sync_persistence(cls) -> SyncPersistenceProtocol[T]: