Fix pre-commit errors

triton-inference-server · Oct 20, 2023 · 7e8cd1b · 7e8cd1b
1 parent fe81e30
commit 7e8cd1b
Show file tree

Hide file tree

Showing 7 changed files with 12 additions and 11 deletions.
diff --git a/ci/L0_accuracy_test/test.sh b/ci/L0_accuracy_test/test.sh
diff --git a/ci/L0_accuracy_test/vllm_accuracy_test.py b/ci/L0_accuracy_test/vllm_accuracy_test.py
@@ -24,21 +24,20 @@
 # (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
 # OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 
+import asyncio
 import sys
 import unittest
 from functools import partial
 
 import tritonclient.grpc as grpcclient
 from tritonclient.utils import *
-
+from vllm import SamplingParams
 from vllm.engine.arg_utils import AsyncEngineArgs
 from vllm.engine.async_llm_engine import AsyncLLMEngine
 from vllm.utils import random_uuid
-from vllm import SamplingParams
-import asyncio
 
 sys.path.append("../common")
-from test_util import TestResultCollector, create_vllm_request, UserData, callback
+from test_util import TestResultCollector, UserData, callback, create_vllm_request
 
 
 async def generate_python_vllm_output(prompt, llm_engine):

diff --git a/ci/L0_backend_vllm/vllm_backend_test.py b/ci/L0_backend_vllm/vllm_backend_test.py
@@ -33,7 +33,7 @@
 from tritonclient.utils import *
 
 sys.path.append("../common")
-from test_util import TestResultCollector, create_vllm_request, UserData, callback
+from test_util import TestResultCollector, UserData, callback, create_vllm_request
 
 
 class VLLMTritonBackendTest(TestResultCollector):

diff --git a/ci/L0_multi_gpu/test.sh b/ci/L0_multi_gpu/test.sh
diff --git a/ci/L0_multi_gpu/vllm_multi_gpu_test.py b/ci/L0_multi_gpu/vllm_multi_gpu_test.py
@@ -25,14 +25,15 @@
 # OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 
 import sys
+import unittest
+from functools import partial
+
 import nvidia_smi
 import tritonclient.grpc as grpcclient
-from functools import partial
-import unittest
 from tritonclient.utils import *
 
 sys.path.append("../common")
-from test_util import TestResultCollector, create_vllm_request, UserData, callback
+from test_util import TestResultCollector, UserData, callback, create_vllm_request
 
 
 class VLLMMultiGPUTest(TestResultCollector):

diff --git a/ci/L0_stream_enabled/test.sh b/ci/L0_stream_enabled/test.sh
@@ -79,9 +79,9 @@ rm -rf models/
 if [ $RET -eq 1 ]; then
     cat $CLIENT_LOG
     cat $SERVER_LOG
-    echo -e "\n***\n*** Strem Enabled test FAILED. \n***"
+    echo -e "\n***\n*** Straem Enabled test FAILED. \n***"
 else
-    echo -e "\n***\n*** Strem Enabled test PASSED. \n***"
+    echo -e "\n***\n*** Straem Enabled test PASSED. \n***"
 fi
 
 exit $RET
diff --git a/ci/common/test_util.py b/ci/common/test_util.py
@@ -27,8 +27,9 @@
 # OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 
 import json
-import unittest
 import queue
+import unittest
+
 import numpy as np
 import tritonclient.grpc as grpcclient