fix: remove usage of distutils

Partially fixes #2931 distutils is deprecated in Python 3.12 (https://docs.python.org/3/whatsnew/3.10.html#distutils-deprecated) PiperOrigin-RevId: 606697684
googleapis · Feb 13, 2024 · e35ab64 · e35ab64
1 parent d727189
commit e35ab64
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/google/cloud/aiplatform/utils/prediction_utils.py b/google/cloud/aiplatform/utils/prediction_utils.py
@@ -15,7 +15,7 @@
 # limitations under the License.
 #
 
-import distutils.dir_util
+import shutil
 import inspect
 import logging
 import os
@@ -150,4 +150,4 @@ def download_model_artifacts(artifact_uri: str) -> None:
                 blob.download_to_filename(name_without_prefix)
     else:
         # Copy files to the current working directory.
-        distutils.dir_util.copy_tree(artifact_uri, ".")
+        shutil.copytree(artifact_uri, ".")
diff --git a/tests/unit/aiplatform/test_utils.py b/tests/unit/aiplatform/test_utils.py
@@ -139,7 +139,7 @@ def test_invalid_region_does_not_raise_with_valid_region():
 
 @pytest.fixture
 def copy_tree_mock():
-    with mock.patch("distutils.dir_util.copy_tree") as copy_tree_mock:
+    with mock.patch("shutil.copytree") as copy_tree_mock:
         yield copy_tree_mock