Correct return code compairison for yamllint

1 files changed, 5 insertions, 5 deletions

diff --git a/setup.py b/setup.py
index e598c3502..05a23b7bb 100644
--- a/setup.py
+++ b/setup.py
@@ -104,14 +104,14 @@ class OpenShiftAnsibleYamlLint(Command):
                         first = False
 
                     print(format_method(problem, yaml_file))
-                    if problem.level == linter.PROBLEM_LEVELS['error']:
+                    if problem.level == linter.PROBLEM_LEVELS[2]:
                         has_errors = True
-                    elif problem.level == linter.PROBLEM_LEVELS['warning']:
+                    elif problem.level == linter.PROBLEM_LEVELS[1]:
                         has_warnings = True
 
-        assert not has_errors, 'yamllint errors found'
-        assert not has_warnings, 'yamllint warnings found'
-
+        if has_errors or has_warnings:
+            print('yammlint issues found')
+            exit(1)
 
 class OpenShiftAnsiblePylint(PylintCommand):
     ''' Class to override the default behavior of PylintCommand '''