Skip to content

Commit 7005516

Browse files
committed
fix: codestyle issue
1 parent 9f15206 commit 7005516

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

src/sagemaker/pytorch/estimator.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -162,6 +162,7 @@ def _is_nova_recipe(recipe):
162162

163163
return bool(has_nova_model) or bool(has_distillation)
164164

165+
165166
def _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+
181183
def _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]

0 commit comments

Comments
 (0)