def get_aws_env(args):
    """Returns a dictionary of AWS related environment variables based on command line args"""
    env = {}
    env_values = {
        "AWS_PROFILE": args.aws_profile,
        "AWS_ACCESS_KEY_ID": args.aws_access_key_id,
        "AWS_SECRET_ACCESS_KEY": args.aws_secret_access_key,
        "AWS_SESSION_TOKEN": args.aws_session_token,
    }
    for aws_key, aws_val in env_values.items():
        if aws_val:
            env[aws_key] = aws_val
    return env