mypy-boto3-sagemaker

import boto3 from boto3.session import Session from mypy_boto3_sagemaker.client import SagemakerClient def get_sagemaker_client() -> SagemakerClient: # boto3 automatically picks up credentials from environment variables or AWS config. # For local testing without actual AWS calls, ensure 'boto3' is configured. session = Session() client: SagemakerClient = session.client("sagemaker") return client # Example usage (will only run if AWS credentials are set up) if __name__ == "__main__": try: sagemaker_client = get_sagemaker_client() # Access type-hinted methods, e.g., sagemaker_client.list_training_jobs() print(f"Successfully created SageMaker client: {sagemaker_client.meta.region_name}") # Uncomment below to make an actual API call (requires permissions) # response = sagemaker_client.list_training_jobs(MaxResults=1) # print(f"Training jobs: {response.get('TrainingJobSummaries')}") except Exception as e: print(f"Error creating SageMaker client or making call: {e}")