New tests cover the changes:
Does this PR include breaking model changes? If so, Have you added transformation functions?
Does this PR include the addition of new environment variables in the README?
By submitting this pull request, I confirm that my contribution is made under the terms of the Apache 2.0 license.