Skip to content
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.

Commit d16ed07

Browse files
committedJul 27, 2020
Fix flakes issues
1 parent 8d51e40 commit d16ed07

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed
 

‎test/unit/test_default_inference_handler.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -179,6 +179,7 @@ def test_default_output_fn_gpu(inference_handler):
179179

180180
assert "1,2,3\n4,5,6\n".encode("utf-8") == output
181181

182+
182183
def test_eia_default_model_fn(eia_inference_handler):
183184
with mock.patch("sagemaker_pytorch_serving_container.default_pytorch_inference_handler.os") as mock_os:
184185
mock_os.getenv.return_value = "true"

0 commit comments

Comments
 (0)
Please sign in to comment.