def lambda_handler()

in s3-efs/app.py [0:0]

• 7 lines of code
• 2 McCabe index (conditional complexity)

def lambda_handler(event, context):
    
    if not os.path.isdir(model_dir):
        os.system('mkdir /mnt/ml/models/')

    bucket_name = event['Records'][0]['s3']['bucket']['name']
    key = event['Records'][0]['s3']['object']['key']
    
    s3.download_file(bucket_name, key, f'{model_dir}/{key}')
    
    print("ML Model file downloaded!")