Adding NotFound as a flaky error class.

googleapis · Jan 15, 2015 · 381e1f9 · 381e1f9
1 parent 1741bbb
commit 381e1f9
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/regression/regression_utils.py b/regression/regression_utils.py
@@ -18,6 +18,7 @@
 import types
 
 from gcloud import storage
+from gcloud.storage.exceptions import NotFound
 
 
 # From shell environ. May be None.
@@ -35,7 +36,7 @@
 class RetryTestsMetaclass(type):
 
     NUM_RETRIES = 2
-    FLAKY_ERROR_CLASSES = (AssertionError,)
+    FLAKY_ERROR_CLASSES = (AssertionError, NotFound)
 
     @staticmethod
     def _wrap_class_attr(class_attr):