get_s3_latest_sagemaker_model_assets_key.py

"""
Get the latest model assets key from resulting sagemaker training session
"""
import boto3

S3 = boto3.client('s3')

def get_latest_model_assets_key(bucket: str, prefix: str):
    model_assets = []
    paginator = S3.get_paginator('list_objects')
    for page in paginator.paginate(Bucket=bucket, Prefix=prefix):
        for item in page['Contents']:
            key = item['Key']
            modified_datetime = item['LastModified']
            model_assets.append((modified_datetime, key))
    latest_model_assets_key = sorted(model_assets, reverse=True)[0][1]
    return latest_model_assets_key
        
bucket = 'my-models-data'
prefix = 'training/assets'

latest_models_asset_key = get_latest_model_assets_key(bucket=bucket, prefix=prefix)