File tree 1 file changed +4
-3
lines changed
src/sagemaker/remote_function/runtime_environment
1 file changed +4
-3
lines changed Original file line number Diff line number Diff line change @@ -65,9 +65,6 @@ def main(sys_args=None):
65
65
conda_env = job_conda_env or os .getenv ("SAGEMAKER_JOB_CONDA_ENV" )
66
66
67
67
RuntimeEnvironmentManager ()._validate_python_version (client_python_version , conda_env )
68
- RuntimeEnvironmentManager ()._validate_sagemaker_pysdk_version (
69
- client_sagemaker_pysdk_version
70
- )
71
68
72
69
user = getpass .getuser ()
73
70
if user != "root" :
@@ -89,6 +86,10 @@ def main(sys_args=None):
89
86
client_python_version , conda_env , dependency_settings
90
87
)
91
88
89
+ RuntimeEnvironmentManager ()._validate_sagemaker_pysdk_version (
90
+ client_sagemaker_pysdk_version
91
+ )
92
+
92
93
exit_code = SUCCESS_EXIT_CODE
93
94
except Exception as e : # pylint: disable=broad-except
94
95
logger .exception ("Error encountered while bootstrapping runtime environment: %s" , e )
You can’t perform that action at this time.
0 commit comments