refactor(loading.py): add try-except block to handle KeyError when removing 'model' from params dictionary

This commit is contained in:
Gabriel Almeida 2023-05-06 08:29:38 -03:00
commit bb41f5af5d

View file

@ -74,7 +74,12 @@ def instantiate_class(node_type: str, base_type: str, params: Dict) -> Any:
return loaded_toolkit
elif base_type == "embeddings":
# ? Why remove model from params?
params.pop("model")
try:
params.pop("model")
except KeyError:
pass
# remove all params that are not in class_object.__fields__
try:
return class_object(**params)