Issue #, if available:

Description of changes:

• Raise on NaN val if raise_on_model_failure=True
• Improve return type hint for load_json which can return a list or a dict.

By submitting this pull request, I confirm that you can use, modify, copy, and redistribute this contribution, under the terms of your choice.