experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Merge pull request #40737 from geetachavan1/cherrypicks_45UX9

Browse Source 
[CherryPick 2.3] Skip data loading error in multi_worker_tutorial_test (the test does not aim to cover this).
This commit is contained in:
Goldie Gadde 2020-06-23 17:09:48 -07:00 committed by GitHub
commit fa305ce9a7
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tensorflow/python/keras/distribute/multi_worker_tutorial_test.py
View File

@ -19,6 +19,7 @@ from __future__ import print_function
import contextlib import contextlib
import os import os
import re import re
import zipfile
from absl.testing import parameterized from absl.testing import parameterized
import numpy as np import numpy as np
from tensorflow.python import keras from tensorflow.python import keras
@ -43,6 +44,8 @@ class MultiWorkerTutorialTest(parameterized.TestCase, test.TestCase):
def skip_fetch_failure_exception(self): def skip_fetch_failure_exception(self):
try: try:
yield 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 except Exception as e: # pylint: disable=broad-except
if 'URL fetch failure' in str(e): if 'URL fetch failure' in str(e):
self.skipTest('URL fetch error not considered failure of the test.') self.skipTest('URL fetch error not considered failure of the test.')