You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
27 lines
439 B
27 lines
439 B
from swarm_models import __all__
|
|
|
|
EXPECTED_ALL = [
|
|
"Anthropic",
|
|
"Petals",
|
|
"Mistral",
|
|
"OpenAI",
|
|
"AzureOpenAI",
|
|
"OpenAIChat",
|
|
"Zephyr",
|
|
"Idefics",
|
|
# "Kosmos",
|
|
"Vilt",
|
|
"Nougat",
|
|
"LayoutLMDocumentQA",
|
|
"BioGPT",
|
|
"HuggingfaceLLM",
|
|
"MPT7B",
|
|
"WizardLLMStoryTeller",
|
|
# "GPT4Vision",
|
|
# "Dalle3",
|
|
]
|
|
|
|
|
|
def test_all_imports() -> None:
|
|
assert set(__all__) == set(EXPECTED_ALL)
|