Remove ModuleNotFoundError.

tensorflow · Sep 29, 2019 · 5021681 · 5021681
1 parent 5501ebc
commit 5021681
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/adanet/tf_compat/__init__.py b/adanet/tf_compat/__init__.py
@@ -201,7 +201,7 @@ def test_wrapper(*args, **kwargs):
       _ = tf.contrib
       _ = tf.contrib.estimator.regression_head(
           loss_reduction=tf.compat.v1.losses.Reduction.SUM_OVER_BATCH_SIZE)
-    except (AttributeError, ImportError, ModuleNotFoundError):
+    except (AttributeError, ImportError):
       self.skipTest("Skipping test in TF 2.0.")
     return f(*args, **kwargs)
 
@@ -219,7 +219,7 @@ def test_wrapper(*args, **kwargs):
     try:
       # If tf.contrib doesn't exist, we are in TF 2.0.
       _ = tf_v2.contrib
-    except (AttributeError, ImportError, ModuleNotFoundError):
+    except (AttributeError, ImportError):
       return f(*args, **kwargs)
     self.skipTest("Skipping test in TF 1.0.")
     return f(*args, **kwargs)