Skip to content

Commit a204e71

Browse files
committed
fix test name
1 parent d1db51f commit a204e71

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

tests/unit/test_handler_service_with_context.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -168,7 +168,7 @@ def test_validate_and_initialize_user_module_transform_fn():
168168
)
169169

170170

171-
def test_validate_and_initialize_user_module_transform_fn():
171+
def test_validate_and_initialize_user_module_transform_fn_race_condition():
172172
os.environ["SAGEMAKER_PROGRAM"] = "inference_tranform_fn.py"
173173
inference_handler = handler_service.HuggingFaceHandlerService()
174174
model_dir = os.path.join(os.getcwd(), "tests/resources/model_transform_fn_with_context")

tests/unit/test_handler_service_without_context.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -156,7 +156,7 @@ def test_validate_and_initialize_user_module_transform_fn():
156156
assert inference_handler.transform_fn("model", "dummy", "application/json", "application/json") == "output dummy"
157157

158158

159-
def test_validate_and_initialize_user_module_transform_fn():
159+
def test_validate_and_initialize_user_module_transform_fn_race_condition():
160160
os.environ["SAGEMAKER_PROGRAM"] = "inference_tranform_fn.py"
161161
inference_handler = handler_service.HuggingFaceHandlerService()
162162
model_dir = os.path.join(os.getcwd(), "tests/resources/model_transform_fn_without_context")

0 commit comments

Comments
 (0)