def __init__()

in support/retro_contest/__init__.py [0:0]

• 14 lines of code
• 1 McCabe index (conditional complexity)

    def __init__(self, env, monitorfile, logfile=None):
        gym.Wrapper.__init__(self, env)
        self.file = open(monitorfile, 'w')
        self.csv = csv.DictWriter(self.file, ['r', 'l', 't'])
        self.log = open(logfile, 'w')
        self.logcsv = csv.DictWriter(self.log, ['l', 't'])
        self.episode_reward = 0
        self.episode_length = 0
        self.total_length = 0
        self.start = None
        self.csv.writeheader()
        self.file.flush()
        self.logcsv.writeheader()
        self.log.flush()