applied style

Author: philschmid

src/sagemaker_huggingface_inference_toolkit/decoder_encoder.py

@@ -11,24 +11,17 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
+import csv
 import datetime
 import json
 from io import StringIO
-import csv
+
 import numpy as np
-from mms.service import PredictionException
+from sagemaker_inference import content_types, errors
+from sagemaker_inference.decoder import _npy_to_numpy, _npz_to_sparse
+from sagemaker_inference.encoder import _array_to_npy
 
-from sagemaker_inference.decoder import (
-    _npy_to_numpy,
-    _npz_to_sparse,
-)
-from sagemaker_inference.encoder import (
-    _array_to_npy,
-)
-from sagemaker_inference import (
-    content_types,
-    errors,
-)
+from mms.service import PredictionException
 
 
 def decode_json(content):

src/sagemaker_huggingface_inference_toolkit/handler_service.py

@@ -15,15 +15,15 @@
 import importlib
 import logging
 import os
-import time
 import sys
+import time
 from abc import ABC
 
-from sagemaker_inference import environment, utils, content_types
+from sagemaker_inference import content_types, environment, utils
 from transformers.pipelines import SUPPORTED_TASKS
 
-from mms.service import PredictionException
 from mms import metrics
+from mms.service import PredictionException
 from sagemaker_huggingface_inference_toolkit import decoder_encoder
 from sagemaker_huggingface_inference_toolkit.transformers_utils import (
     _is_gpu_available,

tests/unit/test_decoder_encoder.py

@@ -12,9 +12,11 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 import json
+
 import pytest
-from sagemaker_huggingface_inference_toolkit import decoder_encoder
+
 from mms.service import PredictionException
+from sagemaker_huggingface_inference_toolkit import decoder_encoder
 
 
 ENCODE_JSON_INPUT = {"upper": [1425], "lower": [576], "level": [2], "datetime": ["2012-08-08 15:30"]}

tests/unit/test_handler_service.py

@@ -17,13 +17,14 @@
 import tempfile
 
 import pytest
+from sagemaker_inference import content_types
 from transformers.testing_utils import require_torch, slow
 
 from mms.context import Context, RequestProcessor
 from mms.metrics.metrics_store import MetricsStore
 from sagemaker_huggingface_inference_toolkit import handler_service
 from sagemaker_huggingface_inference_toolkit.transformers_utils import _load_model_from_hub, get_pipeline
-from sagemaker_inference import content_types
+
 
 TASK = "text-classification"
 MODEL = "sshleifer/tiny-dbmdz-bert-large-cased-finetuned-conll03-english"