circular dependency and truncated output

Author: Mohamed Zeidan

src/sagemaker/experiments/run.py

@@ -757,9 +757,28 @@ def __exit__(self, exc_type, exc_value, exc_traceback):
         end_time = datetime.datetime.now(dateutil.tz.tzlocal())
         self._trial_component.end_time = end_time
         if exc_value:
+            # Sanitize and truncate error message to comply with AWS API constraints
+            error_message = str(exc_value)
+            
+            # Be very restrictive - only keep alphanumeric, spaces, and minimal punctuation
+            import re
+            # Remove all problematic characters, keep only basic safe characters
+            error_message = re.sub(r'[^a-zA-Z0-9\s\.\-\,\:\;]', ' ', error_message)
+            
+            # Remove multiple spaces and clean up
+            error_message = re.sub(r'\s+', ' ', error_message).strip()
+            
+            # Truncate to comply with AWS character limit (be more conservative)
+            if len(error_message) > 500:
+                error_message = error_message[:500] + "..."
+                
+            # Fallback to generic message if still problematic
+            if not error_message or len(error_message) < 10:
+                error_message = "Training job failed with execution error"
+                
             self._trial_component.status = _api_types.TrialComponentStatus(
                 primary_status=_TrialComponentStatusType.Failed.value,
-                message=str(exc_value),
+                message=error_message,
             )
         else:
             self._trial_component.status = _api_types.TrialComponentStatus(

src/sagemaker/telemetry/telemetry_logging.py

@@ -18,10 +18,8 @@
 from time import perf_counter
 from typing import List
 import functools
-import requests
 
 import boto3
-from sagemaker.session import Session
 from sagemaker.utils import resolve_value_from_config
 from sagemaker.config.config_schema import TELEMETRY_OPT_OUT_PATH
 from sagemaker.telemetry.constants import (
@@ -186,7 +184,7 @@ def wrapper(*args, **kwargs):
 def _send_telemetry_request(
     status: int,
     feature_list: List[int],
-    session: Session,
+    session,
     failure_reason: str = None,
     failure_type: str = None,
     extra_info: str = None,
@@ -254,8 +252,12 @@ def _requests_helper(url, timeout):
 
     response = None
     try:
+        # Import requests conditionally to avoid breaking imports when requests is not available
+        import requests
         response = requests.get(url, timeout)
-    except requests.exceptions.RequestException as e:
+    except ImportError:
+        logger.debug("requests module not available, skipping telemetry request")
+    except Exception as e:  # This covers requests.exceptions.RequestException and others
         logger.exception("Request exception: %s", str(e))
     return response
 
@@ -281,8 +283,12 @@ def _get_region_or_default(session):
 
 def _get_default_sagemaker_session():
     """Return the default sagemaker session"""
-
-    boto_session = boto3.Session(region_name=DEFAULT_AWS_REGION)
-    sagemaker_session = Session(boto_session=boto_session)
-
-    return sagemaker_session
+    try:
+        # Import Session conditionally to avoid circular import issues
+        from sagemaker.session import Session
+        boto_session = boto3.Session(region_name=DEFAULT_AWS_REGION)
+        sagemaker_session = Session(boto_session=boto_session)
+        return sagemaker_session
+    except ImportError:
+        logger.debug("Session module not available, returning None")
+        return None

tests/data/experiment/train_job_script_for_run_clz.py

@@ -14,9 +14,18 @@
 from __future__ import absolute_import
 
 import os
+import glob
 
-sdk_file = "sagemaker-dev-1.0.tar.gz"
-os.system(f"pip install {sdk_file}")
+# Find the SDK tar file dynamically instead of hardcoding the name
+sdk_files = glob.glob("sagemaker-*.tar.gz")
+if sdk_files:
+    sdk_file = sdk_files[0]  # Use the first (and likely only) SDK tar file found
+    print(f"Installing SDK from: {sdk_file}")
+    os.system(f"pip install {sdk_file}")
+else:
+    print("No SDK tar file found, attempting to install from current directory")
+    # Fallback: try to install the current directory as editable
+    os.system("pip install -e .")
 
 import json
 import logging