build: update isort known first party config (#50)

worker/tests/python/integration/api_server/models/mock_disaggregated_serving/1/model.py

@@ -23,6 +23,7 @@ import uuid
 
 import triton_python_backend_utils as pb_utils
 import ucp
+
 from triton_distributed.icp.nats_request_plane import NatsRequestPlane
 from triton_distributed.icp.ucp_data_plane import UcpDataPlane
 from triton_distributed.worker.remote_operator import RemoteOperator

worker/tests/python/integration/operators/add_multiply_divide.py

@@ -16,6 +16,7 @@
 import asyncio
 
 import numpy
+
 from triton_distributed.worker import Operator, RemoteInferenceRequest, RemoteOperator
 
 

worker/tests/python/integration/operators/identity.py

@@ -15,6 +15,7 @@
 
 
 import numpy
+
 from triton_distributed.worker import Operator, RemoteInferenceRequest
 
 

worker/tests/python/integration/operators/mock_disaggregated_serving.py

@@ -14,10 +14,11 @@
 # limitations under the License.
 
 
+from tritonserver import TritonError
+
 from triton_distributed.worker.operator import Operator
 from triton_distributed.worker.remote_operator import RemoteOperator
 from triton_distributed.worker.remote_request import RemoteInferenceRequest
-from tritonserver import TritonError
 
 
 class MockDisaggregatedServing(Operator):

worker/tests/python/integration/test_add_multiply_divide.py

@@ -23,6 +23,7 @@ import numpy
 import pytest
 import ucp
 from cupy_backends.cuda.api.runtime import CUDARuntimeError
+
 from triton_distributed.icp.nats_request_plane import NatsRequestPlane
 from triton_distributed.icp.ucp_data_plane import UcpDataPlane
 from triton_distributed.worker.deployment import Deployment

worker/tests/python/integration/test_direct.py

@@ -24,6 +24,7 @@ import numpy
 import pytest
 import ucp
 from cupy_backends.cuda.api.runtime import CUDARuntimeError
+
 from triton_distributed.icp.nats_request_plane import NatsRequestPlane
 from triton_distributed.icp.ucp_data_plane import UcpDataPlane
 from triton_distributed.worker.deployment import Deployment

worker/tests/python/integration/test_mock_disaggregated_serving.py

@@ -28,6 +28,9 @@ import tritonclient.grpc as grpcclient
 import ucp
 from cupy_backends.cuda.api.runtime import CUDARuntimeError
 from transformers import XLNetTokenizer
+from tritonclient.utils import InferenceServerException
+from tritonserver import Tensor
+
 from triton_distributed.icp.nats_request_plane import NatsRequestPlane
 from triton_distributed.icp.ucp_data_plane import UcpDataPlane
 from triton_distributed.worker.deployment import Deployment
@@ -36,8 +39,6 @@ from triton_distributed.worker.operator import OperatorConfig
 from triton_distributed.worker.remote_operator import RemoteOperator
 from triton_distributed.worker.triton_core_operator import TritonCoreOperator
 from triton_distributed.worker.worker import WorkerConfig
-from tritonclient.utils import InferenceServerException
-from tritonserver import Tensor
 
 NATS_PORT = 4223
 MODEL_REPOSITORY = (

worker/tests/python/integration/test_perf_benchmark.py

@@ -31,6 +31,7 @@ import logging
 import numpy
 import pytest
 import ucp
+
 from triton_distributed.icp.nats_request_plane import NatsRequestPlane
 from triton_distributed.icp.ucp_data_plane import UcpDataPlane
 from triton_distributed.worker.deployment import Deployment

worker/tests/python/unit/test_args.py

@@ -14,6 +14,7 @@
 # limitations under the License.
 
 import pytest
+
 from triton_distributed.worker.parser import Parser
 
 """

worker/tests/python/unit/test_logger.py

@@ -16,6 +16,7 @@
 import logging
 
 import pytest
+
 from triton_distributed.worker.log_formatter import LOGGER_NAME, setup_logger
 
 logger = logging.getLogger(LOGGER_NAME)