fix: training job name to match ease regex

Author: Namrata Madan

src/sagemaker/remote_function/job.py

@@ -14,6 +14,7 @@
 from __future__ import absolute_import
 
 import os
+import re
 from typing import Dict, List, Tuple
 
 from sagemaker.session import get_execution_role, _logs_for_job, Session
@@ -125,7 +126,11 @@ def start(job_settings: _JobSettings, func, func_args, func_kwargs):
 
         Returns: the _Job object.
         """
-        job_name = name_from_base(func.__name__)
+        # remove all special characters in the beginning of function name
+        job_name_prefix = re.sub(r"^[^a-zA-Z0-9]+", "", func.__name__)
+        # convert all remaining special characters to '-'
+        job_name_prefix = re.sub(r"[^a-zA-Z0-9-]", "-", job_name_prefix)
+        job_name = name_from_base(job_name_prefix)
 
         s3_base_uri = s3_path_join(job_settings.s3_root_uri, job_name)
 

tests/unit/sagemaker/remote_function/test_job.py

@@ -106,7 +106,7 @@ def test_start(session, mock_stored_function, mock_runtime_manager):
 
     job = _Job.start(job_settings, job_function, func_args=(1, 2), func_kwargs={"c": 3, "d": 4})
 
-    assert job.job_name.startswith("job_function")
+    assert job.job_name.startswith("job-function")
 
     assert mock_stored_function.called_once_with(
         sagemaker_session=session(), s3_base_uri=f"{S3_URI}/{job.job_name}", s3_kms_key=None
@@ -168,7 +168,7 @@ def test_start_with_complete_job_settings(session, mock_stored_function, mock_ru
 
     job = _Job.start(job_settings, job_function, func_args=(1, 2), func_kwargs={"c": 3, "d": 4})
 
-    assert job.job_name.startswith("job_function")
+    assert job.job_name.startswith("job-function")
 
     assert mock_stored_function.called_once_with(
         sagemaker_session=session(), s3_base_uri=f"{S3_URI}/{job.job_name}", s3_kms_key=None