remove six (#5333)

Author: Jyothirmaikottu

test/dlc_tests/container_tests/bin/tensorflow_serving/example/mnist_input_data.py

@@ -23,7 +23,7 @@
 import os
 
 import numpy
-from six.moves import urllib
+import urllib.request
 
 # CVDF mirror of http://yann.lecun.com/exdb/mnist/
 SOURCE_URL = "https://storage.googleapis.com/cvdf-datasets/mnist/"

test/dlc_tests/eks/pytorch/training/test_eks_pytorch_training.py

@@ -38,7 +38,7 @@ def test_eks_pytorch_single_node_training(pytorch_training):
     mnist_dataset_download_config = """
       FILE=new_main.py &&
       echo "from __future__ import print_function" > $FILE &&
-      echo "from six.moves import urllib" >> $FILE &&
+      echo "import urllib.request" >> $FILE &&
       echo "from packaging.version import Version" >> $FILE &&
       echo "opener = urllib.request.build_opener()" >> $FILE &&
       echo "opener.addheaders = [('User-agent', 'Mozilla/5.0')]" >> $FILE &&

test/sagemaker_tests/huggingface_tensorflow/training/requirements.txt

@@ -16,7 +16,6 @@ pytest-rerunfailures
 pytest-xdist
 requests
 mock
-six
 python-dateutil>=2.1,<2.8.1
 requests-mock
 # test pipemode fails for 2.x containers with module 'tensorflow' has no attribute 'train'

test/sagemaker_tests/pytorch/inference/requirements.txt

@@ -19,7 +19,6 @@ requests_mock
 retrying==1.3.3
 sagemaker>=2,<3
 sagemaker-inference
-six
 tenacity
 toml
 torch<2.5.0

test/sagemaker_tests/pytorch/training/integration/sagemaker/test_distributed_operations.py

@@ -21,7 +21,7 @@
 from sagemaker.instance_group import InstanceGroup
 from sagemaker.pytorch import PyTorch
 from sagemaker import Session
-from six.moves.urllib.parse import urlparse
+from urllib.parse import urlparse
 from test.test_utils import get_framework_and_version_from_tag, get_cuda_version_from_tag
 from packaging.version import Version
 from packaging.specifiers import SpecifierSet

test/sagemaker_tests/pytorch/training/resources/mnist/mnist.py

@@ -13,7 +13,8 @@
 
 # Workaround for https://github.com/pytorch/vision/issues/1938
 from __future__ import print_function, absolute_import
-from six.moves import urllib
+import urllib.request
+
 
 opener = urllib.request.build_opener()
 opener.addheaders = [("User-agent", "Mozilla/5.0")]

test/sagemaker_tests/pytorch/training/resources/mnist/smdataparallel_mnist.py

@@ -9,7 +9,7 @@
 from __future__ import print_function
 
 # Hack to add mnist dataset download https://github.com/pytorch/vision/issues/3497
-from six.moves import urllib
+import urllib.request
 
 opener = urllib.request.build_opener()
 opener.addheaders = [("User-agent", "Mozilla/5.0")]

test/sagemaker_tests/pytorch/training/resources/mnist/smdebug_mnist.py

@@ -13,7 +13,7 @@
 
 # Workaround for https://github.com/pytorch/vision/issues/1938
 from __future__ import print_function, absolute_import
-from six.moves import urllib
+import urllib.request
 
 opener = urllib.request.build_opener()
 opener.addheaders = [("User-agent", "Mozilla/5.0")]

test/sagemaker_tests/pytorch/training/resources/mnist/smppy_mnist.py

@@ -14,7 +14,7 @@
 # Workaround for https://github.com/pytorch/vision/issues/1938
 from __future__ import absolute_import, print_function
 
-from six.moves import urllib
+import urllib.request
 
 opener = urllib.request.build_opener()
 opener.addheaders = [("User-agent", "Mozilla/5.0")]

test/sagemaker_tests/tensorflow/tensorflow1_training/integration/sagemaker/test_mnist.py

@@ -18,7 +18,7 @@
 import pytest
 from sagemaker.tensorflow import TensorFlow
 from sagemaker.tuner import HyperparameterTuner, IntegerParameter
-from six.moves.urllib.parse import urlparse
+from urllib.parse import urlparse
 
 from ..... import invoke_sm_helper_function
 from test.test_utils import is_pr_context, SKIP_PR_REASON