def test_export_saved_model(self): checkpoint_path = self.save_checkpoint() destination_dir = os.path.join(checkpoint_path, 'export') export_saved_model.export(checkpoint_path, destination_dir, use_tf_sampling=True) self.assertTrue(tf.gfile.Exists( os.path.join(destination_dir, 'saved_model.pb'))) tf.gfile.DeleteRecursively(checkpoint_path)