File tree 2 files changed +3
-3
lines changed
google/cloud/aiplatform/utils
2 files changed +3
-3
lines changed Original file line number Diff line number Diff line change 15
15
# limitations under the License.
16
16
#
17
17
18
- import distutils . dir_util
18
+ import shutil
19
19
import inspect
20
20
import logging
21
21
import os
@@ -150,4 +150,4 @@ def download_model_artifacts(artifact_uri: str) -> None:
150
150
blob .download_to_filename (name_without_prefix )
151
151
else :
152
152
# Copy files to the current working directory.
153
- distutils . dir_util . copy_tree (artifact_uri , "." )
153
+ shutil . copytree (artifact_uri , "." )
Original file line number Diff line number Diff line change @@ -139,7 +139,7 @@ def test_invalid_region_does_not_raise_with_valid_region():
139
139
140
140
@pytest .fixture
141
141
def copy_tree_mock ():
142
- with mock .patch ("distutils.dir_util.copy_tree " ) as copy_tree_mock :
142
+ with mock .patch ("shutil.copytree " ) as copy_tree_mock :
143
143
yield copy_tree_mock
144
144
145
145
You can’t perform that action at this time.
0 commit comments