Trying to fix lldb build breakage probably caused by rL282452

llvm-svn: 282479

1 files changed, 11 insertions, 11 deletions

diff --git a/llvm/utils/lit/lit/run.py b/llvm/utils/lit/lit/run.py
index 6bcca012a5f..49569084926 100644
--- a/llvm/utils/lit/lit/run.py
+++ b/llvm/utils/lit/lit/run.py
@@ -151,19 +151,19 @@ def run_one_tester(run, provider, display):
     tester.run()
 
 ###
+class _Display(object):
+    def __init__(self, display, provider, maxFailures):
+        self.display = display
+        self.maxFailures = maxFailures or object()
+        self.failedCount = 0
+    def update(self, test):
+        self.display.update(test)
+        self.failedCount += (test.result.code == lit.Test.FAIL)
+        if self.failedCount == self.maxFailures:
+            provider.cancel()
 
 def handleFailures(provider, consumer, maxFailures):
-    class _Display(object):
-        def __init__(self, display):
-            self.display = display
-            self.maxFailures = maxFailures or object()
-            self.failedCount = 0
-        def update(self, test):
-            self.display.update(test)
-            self.failedCount += (test.result.code == lit.Test.FAIL)
-            if self.failedCount == self.maxFailures:
-                provider.cancel()
-    consumer.display = _Display(consumer.display)
+    consumer.display = _Display(consumer.display, provider, maxFailures)
 
 class Run(object):
     """