Evan Rosen

Thanks for bringing this to our attention. This looks like a bug in some code that is attempting to construct pyarrow.RecordBatches in a special way. I suspect this was intended...

This issue should be fixed at head. Let us know if you are still seeing the problem after https://github.com/tensorflow/model-analysis/commit/d31e583a7a8765b6d89264d24a18b3d133b33b89

Hey Steven, this is something I never got around to implementing. It should be pretty simple however. I'm not in a position to fix this immediately, but I would more...