diff --git a/.github/workflows/python-package.yml b/.github/workflows/python-package.yml
index bfc9753..984cc47 100644
--- a/.github/workflows/python-package.yml
+++ b/.github/workflows/python-package.yml
@@ -34,7 +34,7 @@ jobs:
         # stop the build if there are Python syntax errors
         flake8 . --count --select=E9,F63,F7 --show-source --statistics
         # exit-zero treats all errors as warnings. The GitHub editor is 127 chars wide
-        flake8 . --count --exit-zero --max-complexity=10 --max-line-length=127 --statistics
+        flake8 . --count --max-complexity=10 --max-line-length=127 --statistics
     - name: Test with pytest
       run: |
         pytest
diff --git a/apt/risk/__init__.py b/apt/risk/__init__.py
index 8b13789..e69de29 100644
--- a/apt/risk/__init__.py
+++ b/apt/risk/__init__.py
@@ -1 +0,0 @@
-
diff --git a/apt/risk/data_assessment/dataset_assessment_manager.py b/apt/risk/data_assessment/dataset_assessment_manager.py
index a94fe70..1f17947 100644
--- a/apt/risk/data_assessment/dataset_assessment_manager.py
+++ b/apt/risk/data_assessment/dataset_assessment_manager.py
@@ -78,9 +78,9 @@ class DatasetAssessmentManager:
         if self.config.persist_reports:
             results_log_file = "_results.log.csv"
             self._dump_scores_to_file(self.attack_scores_per_record_knn_probabilities,
-                                     "per_record_knn_probabilities" + results_log_file, True)
+                                      "per_record_knn_probabilities" + results_log_file, True)
             self._dump_scores_to_file(self.attack_scores_whole_dataset_knn_distance,
-                                     "whole_dataset_knn_distance" + results_log_file, True)
+                                      "whole_dataset_knn_distance" + results_log_file, True)
 
     @staticmethod
     def _dump_scores_to_file(attack_scores: list[DatasetAttackScore], filename: str, header: bool):
diff --git a/setup.cfg b/setup.cfg
index 77e9de1..d75c434 100644
--- a/setup.cfg
+++ b/setup.cfg
@@ -25,4 +25,8 @@ exclude =
 
 [flake8]
 ignore = C901,W503
-per-file-ignores = __init__.py:F401
\ No newline at end of file
+per-file-ignores = __init__.py:F401
+builtins = keras,xgboost
+exclude =
+    venv
+    venv1
diff --git a/tests/test_data_assessment.py b/tests/test_data_assessment.py
index b83a382..8ecad07 100644
--- a/tests/test_data_assessment.py
+++ b/tests/test_data_assessment.py
@@ -34,9 +34,9 @@ def teardown_function():
 
 
 anon_testdata = [('iris_np', iris_dataset_np, 'np', k, mgr) for k in range(2, 10, 4)] \
-                + [('diabetes_np', diabetes_dataset_np, 'np', k, mgr) for k in range(2, 10, 4)] \
-                + [('nursery_pd', nursery_dataset_pd, 'pd', k, mgr) for k in range(2, 10, 4)] \
-                + [('adult_pd', adult_dataset_pd, 'pd', k, mgr) for k in range(2, 10, 4)]
+    + [('diabetes_np', diabetes_dataset_np, 'np', k, mgr) for k in range(2, 10, 4)] \
+    + [('nursery_pd', nursery_dataset_pd, 'pd', k, mgr) for k in range(2, 10, 4)] \
+    + [('adult_pd', adult_dataset_pd, 'pd', k, mgr) for k in range(2, 10, 4)]
 
 
 @pytest.mark.parametrize("name, data, dataset_type, k, mgr", anon_testdata)
diff --git a/tests/test_data_assessment_short_test.py b/tests/test_data_assessment_short_test.py
index 1089d4c..0b575c0 100644
--- a/tests/test_data_assessment_short_test.py
+++ b/tests/test_data_assessment_short_test.py
@@ -29,9 +29,9 @@ def teardown_function():
 
 
 anon_testdata = [('iris_np', iris_dataset_np, 'np', mgr1)] \
-                + [('nursery_pd', nursery_dataset_pd, 'pd', mgr2)] \
-                + [('iris_np', iris_dataset_np, 'np', mgr3)] \
-                + [('nursery_pd', nursery_dataset_pd, 'pd', mgr4)]
+    + [('nursery_pd', nursery_dataset_pd, 'pd', mgr2)] \
+    + [('iris_np', iris_dataset_np, 'np', mgr3)] \
+    + [('nursery_pd', nursery_dataset_pd, 'pd', mgr4)]
 
 
 @pytest.mark.parametrize("name, data, dataset_type, mgr", anon_testdata)