def test_search_sparse_recipes_from_stub(model_stub, other_args):
model = Zoo.load_model_from_stub(model_stub, **other_args)
recipes = Zoo.search_sparse_recipes(model_stub)
assert len(recipes) > 0
for recipe in recipes:
assert recipe
assert recipe.model_metadata.domain == model.domain
assert recipe.model_metadata.sub_domain == model.sub_domain
assert recipe.model_metadata.architecture == model.architecture
assert recipe.model_metadata.sub_architecture == model.sub_architecture
assert recipe.model_metadata.framework == model.framework
assert recipe.model_metadata.repo == model.repo
assert recipe.model_metadata.dataset == model.dataset
assert recipe.model_metadata.training_scheme == model.training_scheme
def test_search_sparse_recipes(model_args, other_args, other_recipe_args):
model = Zoo.load_model(**model_args, **other_args)
recipes = Zoo.search_sparse_recipes(model, **other_recipe_args)
assert len(recipes) > 0
for recipe in recipes:
assert recipe
assert recipe.model_metadata.domain == model.domain
assert recipe.model_metadata.sub_domain == model.sub_domain
assert recipe.model_metadata.architecture == model.architecture
assert recipe.model_metadata.sub_architecture == model.sub_architecture
assert recipe.model_metadata.framework == model.framework
assert recipe.model_metadata.repo == model.repo
assert recipe.model_metadata.dataset == model.dataset
assert recipe.model_metadata.training_scheme == model.training_scheme
if "recipe_type" in other_recipe_args:
assert recipe.recipe_type == other_recipe_args["recipe_type"]