def close(self):
    if hasattr(self.env, 'close'):
        self.env.close()
    tf.get_default_session().__exit__(None, None, None)
    logger.reset()
def close(self):
    if hasattr(self.env, 'close'):
        self.env.close()
    tf.get_default_session().__exit__(None, None, None)
    logger.reset()