'Refactored by Sourcery'

litestar-org · Jan 9, 2024 · 84fdf9f · 84fdf9f
1 parent 45c3e3e
commit 84fdf9f
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/polyfactory/factories/base.py b/polyfactory/factories/base.py
@@ -241,10 +241,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]: