From a383e26bd9e29ef65f8bdbda5a52637a77486783 Mon Sep 17 00:00:00 2001 From: Sourcery AI <> Date: Mon, 15 Jan 2024 11:32:55 +0000 Subject: [PATCH] 'Refactored by Sourcery' --- polyfactory/factories/base.py | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) diff --git a/polyfactory/factories/base.py b/polyfactory/factories/base.py index d49692f3..767fe815 100644 --- a/polyfactory/factories/base.py +++ b/polyfactory/factories/base.py @@ -245,10 +245,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]: