Simplify model initialization

This commit is contained in:
Andrea Lanfranchi 2024-02-21 15:18:51 +01:00
parent adf6328cd9
commit 810dfac4fe
No known key found for this signature in database
GPG Key ID: 2D623A60BB279915

View File

@ -43,8 +43,7 @@ def build_model(model_name: str) -> Any:
if not model_name in model_obj.keys():
model = models.get(model_name)
if model:
model = model()
model_obj[model_name] = model
model_obj[model_name] = model()
else:
raise ValueError(f"Invalid model_name passed - {model_name}")