Merge pull request #1007 from drikusroor/fix-flake8-issues

Fix flake8 issues

.github/workflows/ci.yml

@@ -31,8 +31,8 @@ jobs:
         pip install -r requirements.txt
 
     - name: Lint with flake8
-      continue-on-error: true
+      continue-on-error: false
-      run: flake8 scripts/ tests/
+      run: flake8 scripts/ tests/ --select E303,W293,W291,W292,E305
 
     - name: Run unittest tests with coverage
       run: |

README.md

@@ -341,4 +341,7 @@ This project uses [flake8](https://flake8.pycqa.org/en/latest/) for linting. To
 
 ```
 flake8 scripts/ tests/
+
+# Or, if you want to run flake8 with the same configuration as the CI:
+flake8 scripts/ tests/ --select E303,W293,W291,W292,E305
 ```

scripts/ai_functions.py

@@ -45,8 +45,6 @@ def improve_code(suggestions: List[str], code: str) -> str:
     result_string = call_ai_function(function_string, args, description_string)
     return result_string
 
-
-
 def write_tests(code: str, focus: List[str]) -> str:
     """
     A function that takes in code and focus topics and returns a response from create chat completion api call.

scripts/logger.py

@@ -159,6 +159,7 @@ class ConsoleHandler(logging.StreamHandler):
         except Exception:
             self.handleError(record)
 
+
 '''
 Allows to handle custom placeholders 'title_color' and 'message_no_color'.
 To use this formatter, make sure to pass 'color', 'title' as log extras.

scripts/memory/__init__.py

@@ -44,6 +44,7 @@ def get_memory(cfg, init=False):
 def get_supported_memory_backends():
     return supported_memory
 
+
 __all__ = [
     "get_memory",
     "LocalCache",

tests/integration/memory_tests.py

@@ -45,5 +45,6 @@ class TestLocalCache(unittest.TestCase):
         self.assertEqual(len(relevant_texts), k)
         self.assertIn(self.example_texts[1], relevant_texts)
 
+
 if __name__ == '__main__':
     unittest.main()

tests/local_cache_test.py

@@ -47,5 +47,6 @@ class TestLocalCache(unittest.TestCase):
         stats = self.cache.get_stats()
         self.assertEqual(stats, (1, self.cache.data.embeddings.shape))
 
+
 if __name__ == '__main__':
     unittest.main()

tests/test_browse_scrape_text.py

@@ -33,8 +33,6 @@ Additional aspects:
 - The function uses a generator expression to split the text into lines and chunks, which can improve performance for large amounts of text.
 """
 
-
-
 class TestScrapeText:
 
     # Tests that scrape_text() returns the expected text when given a valid URL.

tests/test_json_parser.py

@@ -66,8 +66,6 @@ class TestParseJson(unittest.TestCase):
         # Assert that this raises an exception:
         self.assertEqual(fix_and_parse_json(json_str, try_to_fix_with_gpt=False), good_obj)
 
-
-
     def test_invalid_json_leading_sentence_with_gpt(self):
         # Test that a REALLY invalid JSON string raises an error when try_to_fix_with_gpt is False
         json_str = """I will first need to browse the repository (https://github.com/Torantulino/Auto-GPT) and identify any potential bugs that need fixing. I will use the "browse_website" command for this.
@@ -108,6 +106,5 @@ class TestParseJson(unittest.TestCase):
         self.assertEqual(fix_and_parse_json(json_str, try_to_fix_with_gpt=False), good_obj)
 
 
-
 if __name__ == '__main__':
     unittest.main()

tests/unit/json_tests.py

@@ -68,8 +68,6 @@ class TestParseJson(unittest.TestCase):
         # Assert that this raises an exception:
         self.assertEqual(fix_and_parse_json(json_str, try_to_fix_with_gpt=False), good_obj)
 
-
-
     def test_invalid_json_leading_sentence_with_gpt(self):
         # Test that a REALLY invalid JSON string raises an error when try_to_fix_with_gpt is False
         json_str = """I will first need to browse the repository (https://github.com/Torantulino/Auto-GPT) and identify any potential bugs that need fixing. I will use the "browse_website" command for this.
@@ -110,6 +108,5 @@ class TestParseJson(unittest.TestCase):
         self.assertEqual(fix_and_parse_json(json_str, try_to_fix_with_gpt=False), good_obj)
 
 
-
 if __name__ == '__main__':
     unittest.main()