def get_deepobs_dataloader(problem_cls, eval_batchsize):
    """Get training dataloader of DeepOBS problems."""
    problem = problem_cls(batch_size=eval_batchsize)
    problem.set_up()

    # Initialize the testproblem instance to train mode and return dataloader
    problem.train_init_op()
    return problem.data._train_dataloader