Issue #, if available:

Description of changes: This PR handles the case where predictions are already DatasetDict. Previously, it broke for univariate tasks if the predictionswere alreadyDatasetDict`.

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