def extra_keyboard_callback()

in rllib_env_imitation.py [0:0]

• 38 lines of code
• 11 McCabe index (conditional complexity)

    def extra_keyboard_callback(self, key):
        if key == b'r':
            s = self.env.reset()
        elif key == b'R':
            s = self.env.reset(start_time=0.0)
        elif key == b' ':
            self.time_checker_auto_play.begin()
            self.one_step()
        elif key == b'e':
            self.explore = not self.explore
            print('Exploration:', self.explore)
        elif key == b'c':
            ''' Read a directory for saving images and try to create it '''
            subdir = input("Enter subdirectory for screenshot file: ")
            dir = os.path.join("data/screenshot/", subdir)
            try:
                os.makedirs(dir, exist_ok = True)
            except OSError:
                print("Invalid Subdirectory")
                return
            for i in range(1):
                try:
                    os.makedirs(dir, exist_ok = True)
                except OSError:
                    print("Invalid Subdirectory")
                    continue
                cnt_screenshot = 0
                while True:
                    name = 'screenshot_%04d'%(cnt_screenshot)
                    self.one_step()
                    self.render()
                    self.save_screen(dir=dir, name=name)
                    print('\rsave_screen(%4.4f) / %s' % \
                        (self.env.base_env.get_elapsed_time(), os.path.join(dir,name)), end=" ")
                    cnt_screenshot += 1
                    if self.env.base_env._end_of_episode:
                        break
                print("\n")