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"]