Fixed the Gluon tests in GenerationTest
No idea why and how this is working in the pipeline. I get errors for different tests (including TensorFlow tests) locally. But it's working in the pipeline? So maybe it's fixed and it's just my computer acting up again.