Skip to content

Commit

Permalink
Master jumpstart curated hub (aws#4464)
Browse files Browse the repository at this point in the history
  • Loading branch information
bencrabtree committed Mar 15, 2024
1 parent 5bc742f commit 9210e49
Show file tree
Hide file tree
Showing 3 changed files with 6 additions and 0 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -65,6 +65,9 @@ def main(sys_args=None):
conda_env = job_conda_env or os.getenv("SAGEMAKER_JOB_CONDA_ENV")

RuntimeEnvironmentManager()._validate_python_version(client_python_version, conda_env)
RuntimeEnvironmentManager()._validate_sagemaker_pysdk_version(
client_sagemaker_pysdk_version
)

user = getpass.getuser()
if user != "root":
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -24,6 +24,8 @@
import dataclasses
import json

import sagemaker


class _UTCFormatter(logging.Formatter):
"""Class that overrides the default local time provider in log formatter."""
Expand Down
1 change: 1 addition & 0 deletions src/sagemaker/utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -22,6 +22,7 @@
import random
import re
import shutil
import sys
import tarfile
import tempfile
import time
Expand Down

0 comments on commit 9210e49

Please sign in to comment.