From f5369b0d413512688161fb15945cf87f11f8ff6d Mon Sep 17 00:00:00 2001 From: Louis Dupont Date: Tue, 13 Dec 2022 11:52:00 +0200 Subject: [PATCH] done --- src/super_gradients/common/factories/type_factory.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/super_gradients/common/factories/type_factory.py b/src/super_gradients/common/factories/type_factory.py index ef961f09a4..d535b49c09 100644 --- a/src/super_gradients/common/factories/type_factory.py +++ b/src/super_gradients/common/factories/type_factory.py @@ -29,7 +29,7 @@ def get(self, conf: Union[str, type]): If provided value is already a class type, the value will be returned as is. """ - if isinstance(conf, str): + if isinstance(conf, str) or isinstance(conf, bool): if conf in self.type_dict: return self.type_dict[conf] else: