examples/research_projects/rag-end2end-retriever/finetune_rag.py/GenerativeQAModule/test_step
class GenerativeQAModule:    def test_step(self, batch, batch_idx):
        return self._generative_step(batch)