fix test_factory

aiidateam · Jan 17, 2025 · 5347271 · 5347271
1 parent 28d7971
commit 5347271
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/aiida/plugins/factories.py b/src/aiida/plugins/factories.py
@@ -410,7 +410,7 @@ def StorageFactory(entry_point_name: str, load: bool = True) -> Union[EntryPoint
 
 
 @overload
-def TransportFactory(entry_point_name: str, load: Literal[True] = True) -> Union[Type['Transport']]: ...
+def TransportFactory(entry_point_name: str, load: Literal[True] = True) -> Type['Transport']: ...
 
 
 @overload
@@ -430,7 +430,7 @@ def TransportFactory(entry_point_name: str, load: bool = True) -> Union[EntryPoi
 
     entry_point_group = 'aiida.transports'
     entry_point = BaseFactory(entry_point_group, entry_point_name, load=load)
-    valid_classes = Transport
+    valid_classes = (Transport,)
 
     if not load:
         return entry_point