From b6988bd9ae4ed7a2c4af50947466934e4b967b9a Mon Sep 17 00:00:00 2001 From: Sourcery AI <> Date: Tue, 16 Jan 2024 07:05:11 +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 9ab29c0e..41a442d5 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]: