diff --git a/sagemaker-serve/tests/integ/test_in_process_integration.py b/sagemaker-serve/tests/integ/test_in_process_integration.py
index d9705f9977..e218da234f 100644
--- a/sagemaker-serve/tests/integ/test_in_process_integration.py
+++ b/sagemaker-serve/tests/integ/test_in_process_integration.py
@@ -13,7 +13,6 @@
 from __future__ import absolute_import
 
 import uuid
-import pytest
 import logging
 
 from sagemaker.serve.model_builder import ModelBuilder
diff --git a/sagemaker-serve/tests/integ/test_jumpstart_deploy_parity.py b/sagemaker-serve/tests/integ/test_jumpstart_deploy_parity.py
index ab136803b5..94edaaa537 100644
--- a/sagemaker-serve/tests/integ/test_jumpstart_deploy_parity.py
+++ b/sagemaker-serve/tests/integ/test_jumpstart_deploy_parity.py
@@ -12,7 +12,6 @@
 from __future__ import absolute_import
 
 import uuid
-import pytest
 import logging
 
 from sagemaker.serve.model_builder import ModelBuilder
diff --git a/sagemaker-serve/tests/integ/test_jumpstart_vllm_integration.py b/sagemaker-serve/tests/integ/test_jumpstart_vllm_integration.py
index f5e8be8c7e..645dd862f8 100644
--- a/sagemaker-serve/tests/integ/test_jumpstart_vllm_integration.py
+++ b/sagemaker-serve/tests/integ/test_jumpstart_vllm_integration.py
@@ -13,7 +13,6 @@
 from __future__ import absolute_import
 
 import uuid
-import pytest
 import logging
 
 from sagemaker.serve.model_builder import ModelBuilder
diff --git a/sagemaker-serve/tests/integ/test_passthrough_source_code_repack_integration.py b/sagemaker-serve/tests/integ/test_passthrough_source_code_repack_integration.py
index c5c5826335..177698b8dd 100644
--- a/sagemaker-serve/tests/integ/test_passthrough_source_code_repack_integration.py
+++ b/sagemaker-serve/tests/integ/test_passthrough_source_code_repack_integration.py
@@ -26,7 +26,6 @@
 import uuid
 
 import boto3
-import pytest
 
 from sagemaker.serve.model_builder import ModelBuilder
 from sagemaker.serve.mode.function_pointers import Mode
diff --git a/sagemaker-serve/tests/integ/test_train_inference_e2e_integration.py b/sagemaker-serve/tests/integ/test_train_inference_e2e_integration.py
index ec3eb5a48c..ecfc7437e6 100644
--- a/sagemaker-serve/tests/integ/test_train_inference_e2e_integration.py
+++ b/sagemaker-serve/tests/integ/test_train_inference_e2e_integration.py
@@ -16,7 +16,6 @@
 import uuid
 import tempfile
 import os
-import pytest
 import logging
 
 from sagemaker.serve.model_builder import ModelBuilder
diff --git a/sagemaker-serve/tests/integ/test_triton_integration.py b/sagemaker-serve/tests/integ/test_triton_integration.py
index 7082231dc6..c305fe9f7d 100644
--- a/sagemaker-serve/tests/integ/test_triton_integration.py
+++ b/sagemaker-serve/tests/integ/test_triton_integration.py
@@ -16,7 +16,6 @@
 import json
 import uuid
 import tempfile
-import pytest
 import logging
 
 from sagemaker.serve.model_builder import ModelBuilder