def init_environment()

in envs/thor.py [0:0]

• 12 lines of code
• 8 McCabe index (conditional complexity)

    def init_environment(self):

        scene, episode = None, None
        
        if self.config.MODE == 'train':
            scene = self.rs.choice(['FloorPlan%d'%idx for idx in range(6, 30+1)]) # 6 --> 30 = train. 1 --> 5 = test

        elif self.config.MODE == 'eval':
            if not hasattr(self, 'test_episodes'):
                self.test_episodes = iter(self.config.ENV.TEST_EPISODES)
            scene, episode = next(self.test_episodes)
            print ('INIT: %s episode %s'%(scene, episode))
        
        self.init_scene_and_agent(scene=scene, episode=episode)

        # store all reachable x, z for snapping to grid
        self.reachable_x = list(set([pos[0] for pos in self.reachable_positions]))
        self.reachable_z = list(set([pos[2] for pos in self.reachable_positions]))