PR comments

neuralmagic · Sep 15, 2023 · 24cf489 · 24cf489
1 parent cc87c18
commit 24cf489
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/sparsezoo/objects/recipes.py b/src/sparsezoo/objects/recipes.py
@@ -47,7 +47,7 @@ def __init__(
         self._default_recipe_name = self._RECIPE_DEFAULT_NAME
         custom_default = stub_params.get("recipe_type") or stub_params.get("recipe")
         if custom_default is not None:
-            self._default_recipe_name = "recipe_" + custom_default + ".md"
+            self._default_recipe_name = "recipe_" + custom_default
 
     @property
     def recipes(self) -> List:
@@ -62,7 +62,7 @@ def default(self) -> File:
         :return: The default recipe in the recipe list
         """
         for recipe in self._recipes:
-            if recipe.name == self._default_recipe_name:
+            if recipe.name.startswith(self._default_recipe_name):
                 return recipe
 
         # fallback to first recipe in list