fix: ignore type for specific case

This commit is contained in:
Gabriel Almeida 2023-04-02 10:42:24 -03:00
commit 5a2fca7eb5

View file

@ -40,7 +40,7 @@ def import_by_type(_type: str, name: str) -> Any:
}
if _type == "llms":
key = "chat" if "chat" in name.lower() else "llm"
loaded_func = func_dict[_type][key]
loaded_func = func_dict[_type][key] # type: ignore
else:
loaded_func = func_dict[_type]