From 3a8f6c021e2da4859363ba492363a4103627f616 Mon Sep 17 00:00:00 2001 From: Sourcery AI <> Date: Wed, 3 Jan 2024 14:01:46 +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 4ea192dd..25753510 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]: