, because we used some of torchvision source code

also quick fix ucb learner

of AaLearner's (except the parent class AaLearner itself)

Add test for GenLearner