File tree Expand file tree Collapse file tree 1 file changed +3
-1
lines changed Expand file tree Collapse file tree 1 file changed +3
-1
lines changed Original file line number Diff line number Diff line change @@ -162,6 +162,7 @@ def _is_nova_recipe(recipe):
162162
163163 return bool (has_nova_model ) or bool (has_distillation )
164164
165+
165166def _is_eval_recipe (recipe ):
166167 """Check if the recipe is an eval recipe.
167168
@@ -178,6 +179,7 @@ def _is_eval_recipe(recipe):
178179 eval_config = recipe .get ("evaluation" , {})
179180 return bool (eval_config )
180181
182+
181183def _recipe_initialize_args (source_dir ):
182184 """Initialize the arguments dictionary for recipe setup.
183185
@@ -964,7 +966,7 @@ def _device_validate_and_get_type(kwargs, recipe):
964966 if "instance_type" not in kwargs :
965967 raise ValueError ("Must pass instance type to estimator when using training recipes." )
966968
967- if not _is_nova_recipe (recipe ) and "trainer" not in recipe and not _is_eval_recipe (recipe ):
969+ if not _is_nova_recipe (recipe ) and "trainer" not in recipe and not _is_eval_recipe (recipe ):
968970 raise ValueError ("Supplied recipe does not contain required field trainer." )
969971
970972 instance_type = kwargs ["instance_type" ].split ("." )[1 ]
You can’t perform that action at this time.
0 commit comments