未验证 提交 fa305ce9 编写于 作者: G Goldie Gadde 提交者: GitHub

Merge pull request #40737 from geetachavan1/cherrypicks_45UX9

[CherryPick 2.3] Skip data loading error in multi_worker_tutorial_test (the test does not aim to cover this).
......@@ -19,6 +19,7 @@ from __future__ import print_function
import contextlib
import os
import re
import zipfile
from absl.testing import parameterized
import numpy as np
from tensorflow.python import keras
......@@ -43,6 +44,8 @@ class MultiWorkerTutorialTest(parameterized.TestCase, test.TestCase):
def skip_fetch_failure_exception(self):
try:
yield
except zipfile.BadZipfile as e:
self.skipTest('Data loading error: Bad magic number for file header.')
except Exception as e: # pylint: disable=broad-except
if 'URL fetch failure' in str(e):
self.skipTest('URL fetch error not considered failure of the test.')
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册