From 9462d287fa0b56e31aed01299755e48fce31f617 Mon Sep 17 00:00:00 2001 From: Sourcery AI <> Date: Tue, 2 Jan 2024 11:16:42 +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 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]: