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: