def METHOD_NAME(d: datetime) -> datetime:
    # There are two types of datetime objects in Python: naive and aware
    # Assume any dbt snapshot timestamp that is naive is meant to represent UTC
    if d is None:
        return d
    elif is_aware(d):
        return d
    else:
        return d.replace(tzinfo=pytz.UTC)