def load_environment(min_random_move_prob, max_random_move_prob, parse_think): dataset = generate_games(min_random_move_prob, max_random_move_prob) parser = XMLParser(fields=["think", "move"]) rubric = Rubric(funcs=[...], weights=[1.0, 1.0, 0.2, 1.0]) return OthelloEnv(dataset, parser, rubric)