def close()

in data_loaders/generate_tfr/generate.py [0:0]

• 10 lines of code
• 5 McCabe index (conditional complexity)

    def close(self):
        if self.print_progress:
            print('%-40s\r' % 'Flushing data...', end='', flush=True)
        for (writers, _) in self.tfr_writers:
            for writer in writers:
                writer.close()
        self.tfr_writers = []
        if self.print_progress:
            print('%-40s\r' % '', end='', flush=True)
            print('Added %d images.' % self.cur_images)