From 534727185df294935afadac32023c60ddb44af63 Mon Sep 17 00:00:00 2001 From: Ali Date: Fri, 17 Jan 2025 10:51:41 +0100 Subject: [PATCH] fix test_factory --- src/aiida/plugins/factories.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/aiida/plugins/factories.py b/src/aiida/plugins/factories.py index 137920b26..925b40067 100644 --- 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