def test_set_all_curriculums_to_lesson_num(curriculum_a, curriculum_b): meta_curriculum = MetaCurriculum({"Brain1": curriculum_a, "Brain2": curriculum_b}) meta_curriculum.set_all_curricula_to_lesson_num(2) assert curriculum_a.lesson_num == 2 assert curriculum_b.lesson_num == 2
def test_set_all_curriculums_to_lesson_num(): meta_curriculum = MetaCurriculum(test_meta_curriculum_config) meta_curriculum.set_all_curricula_to_lesson_num(2) assert meta_curriculum.brains_to_curricula["Brain1"].lesson_num == 2 assert meta_curriculum.brains_to_curricula["Brain2"].lesson_num == 2
def try_create_meta_curriculum( curriculum_config: Optional[Dict], env: SubprocessEnvManager, lesson: int ) -> Optional[MetaCurriculum]: if curriculum_config is None: return None else: meta_curriculum = MetaCurriculum(curriculum_config) # TODO: Should be able to start learning at different lesson numbers # for each curriculum. meta_curriculum.set_all_curricula_to_lesson_num(lesson) return meta_curriculum