mlem
mlem copied to clipboard
Support combining batched prediction results on DatasetType
Context
A bug was introduced in https://github.com/iterative/mlem/pull/221, where approach to combining prediction results incorrectly assumed type to be numpy array. This PR utilizes DatasetAnalyzer to first determine the type of the results prior to combining them, since the same model can output different data types.
Modifications
mlem/api/commands.py- Interpret data type prior to combining prediction resultsmlem/contrib/numpy.py- Add support for combining batch reading prediction results for nd.array
Which issue(s) this PR fixes: Fixes https://github.com/iterative/mlem/issues/257
I've updated the branch, now auth issues are gone. But there are some others I didn't investigate. @terryyylim do you mind taking a look?