test_transform_batch fails on python3.9

tests/losses/ctc_loss_test.py::test_transform_batch fails on py3.9 See https://gitlab.teklia.com/atr/pylaia/-/jobs/539516#L510 or any recent pipeline