def tearDown(self): if self.resolver: tpu_strategy_util.shutdown_tpu_system(self.resolver) super(TPUEmbeddingCorrectness, self).tearDown()
def tearDown(self): tpu_strategy_util.shutdown_tpu_system(self.resolver) super(TPUEmbeddingCheckpointTest, self).tearDown()