Skip to content

Commit 9210e49

Browse files
committed
Master jumpstart curated hub (aws#4464)
1 parent 5bc742f commit 9210e49

File tree

3 files changed

+6
-0
lines changed

3 files changed

+6
-0
lines changed

src/sagemaker/remote_function/runtime_environment/bootstrap_runtime_environment.py

+3
Original file line numberDiff line numberDiff line change
@@ -65,6 +65,9 @@ def main(sys_args=None):
6565
conda_env = job_conda_env or os.getenv("SAGEMAKER_JOB_CONDA_ENV")
6666

6767
RuntimeEnvironmentManager()._validate_python_version(client_python_version, conda_env)
68+
RuntimeEnvironmentManager()._validate_sagemaker_pysdk_version(
69+
client_sagemaker_pysdk_version
70+
)
6871

6972
user = getpass.getuser()
7073
if user != "root":

src/sagemaker/remote_function/runtime_environment/runtime_environment_manager.py

+2
Original file line numberDiff line numberDiff line change
@@ -24,6 +24,8 @@
2424
import dataclasses
2525
import json
2626

27+
import sagemaker
28+
2729

2830
class _UTCFormatter(logging.Formatter):
2931
"""Class that overrides the default local time provider in log formatter."""

src/sagemaker/utils.py

+1
Original file line numberDiff line numberDiff line change
@@ -22,6 +22,7 @@
2222
import random
2323
import re
2424
import shutil
25+
import sys
2526
import tarfile
2627
import tempfile
2728
import time

0 commit comments

Comments
 (0)