Skip to content

Commit 710d4e9

Browse files
author
Ashish Gupta
committed
fix styling
1 parent e608acf commit 710d4e9

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

tests/unit/sagemaker/image_uris/test_neo.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -117,9 +117,11 @@ def test_inferentia_mxnet(inferentia_mxnet_version):
117117
def test_inferentia_tensorflow(inferentia_tensorflow_version):
118118
_test_inferentia_framework_uris("tensorflow", inferentia_tensorflow_version)
119119

120+
120121
def test_inferentia_pytorch(inferentia_pytorch_version):
121122
_test_inferentia_framework_uris("pytorch", inferentia_pytorch_version)
122123

124+
123125
def _expected_framework_uri(framework, version, region="us-west-2", processor="cpu"):
124126
return expected_uris.framework_uri(
125127
"sagemaker-{}".format(framework),

0 commit comments

Comments
 (0)