use cmaes curriculum

Author: sumedhpendurkar

src/curriculum/ts_curriculum.py

@@ -11,7 +11,7 @@
 from univar_gaussian_teacher import UnivarGaussianTeacher
 
 class TSCurriculum(RWCurriculum):
-    
+
     # TODO: remove init_v from this class, and perhaps put this in bandit class
     def init_v(self, length:int, mean:int, std:float):
         v = np.zeros(length)
@@ -32,7 +32,7 @@ def learn_online(self, planner, nn_model):
         test_solve = 0
         memory = Memory()
 
-        teacher = UnivarGaussianTeacher(batch_size=self._states_per_difficulty, mean=4, std=4)
+        teacher = CMAESTeacher(batch_size=self._states_per_difficulty, mean=4, std=4)
         ## TODO: remove this TMP!
 
         while test_solve < 1: