Output of Batchify() and Unbatchify() should be a namedtuple if input is a namedtuple

lf1-io / padl

Functional deep learning

Apache License 2.0

106 stars 4 forks source link

Output of Batchify() and Unbatchify() should be a namedtuple if input is a namedtuple #75

Open sjrl opened 2 years ago

sjrl commented 2 years ago

This isn't currently valid, because we only return namedtuple's at the very end of the _apply methods. So Batchify and Unbatchify would never see a namedtuple as input.

sjrl commented 2 years ago

Blocked by Issue #285