Replace flake8 with ruff

thekaranacharya · thekaranacharya · commit 0b85ad1f6e6e · 2024-04-22T13:37:29.000-07:00
diff --git a/guardrails/utils/validator_utils.py b/guardrails/utils/validator_utils.py
@@ -1,7 +1,6 @@
-# flake8: noqa
+# ruff: noqa
 """This module contains the constants and utils used by the validator.py."""
 
-
 from typing import Any, Dict, List, Optional, Tuple, Type, Union
 
 from guardrails.utils.safe_get import safe_get
diff --git a/tests/integration_tests/test_assets/entity_extraction/__init__.py b/tests/integration_tests/test_assets/entity_extraction/__init__.py
@@ -1,4 +1,4 @@
-# flake8: noqa: E501
+# ruff: noqa: E501
 import os
 
 from .optional_prompts import (
diff --git a/tests/integration_tests/test_assets/entity_extraction/optional_prompts.py b/tests/integration_tests/test_assets/entity_extraction/optional_prompts.py
@@ -1,4 +1,4 @@
-# flake8: noqa: E501
+# ruff: noqa: E501
 
 OPTIONAL_PROMPT_COMPLETION_MODEL = """
 Given the following document, answer the following questions. If the answer doesn't exist in the document, enter 'None'.
diff --git a/tests/integration_tests/test_assets/entity_extraction/validated_output_filter.py b/tests/integration_tests/test_assets/entity_extraction/validated_output_filter.py
@@ -1,4 +1,4 @@
-# flake8: noqa: E501
+# ruff: noqa: E501
 VALIDATED_OUTPUT_FILTER = {
     "fees": [
         {
diff --git a/tests/integration_tests/test_assets/entity_extraction/validated_output_fix.py b/tests/integration_tests/test_assets/entity_extraction/validated_output_fix.py
@@ -1,4 +1,4 @@
-# flake8: noqa: E501
+# ruff: noqa: E501
 VALIDATED_OUTPUT_FIX = {
     "fees": [
         {
diff --git a/tests/integration_tests/test_assets/entity_extraction/validated_output_noop.py b/tests/integration_tests/test_assets/entity_extraction/validated_output_noop.py
@@ -1,4 +1,4 @@
-# flake8: noqa: E501
+# ruff: noqa: E501
 VALIDATED_OUTPUT_NOOP = {
     "fees": [
         {
diff --git a/tests/integration_tests/test_assets/entity_extraction/validated_output_reask_1.py b/tests/integration_tests/test_assets/entity_extraction/validated_output_reask_1.py
@@ -1,4 +1,4 @@
-# flake8: noqa: E501
+# ruff: noqa: E501
 from guardrails.utils.reask_utils import FieldReAsk
 from guardrails.validators import FailResult
 
diff --git a/tests/integration_tests/test_assets/entity_extraction/validated_output_reask_2.py b/tests/integration_tests/test_assets/entity_extraction/validated_output_reask_2.py
@@ -1,4 +1,4 @@
-# flake8: noqa: E501
+# ruff: noqa: E501
 VALIDATED_OUTPUT_REASK_2 = {
     "fees": [
         {
diff --git a/tests/integration_tests/test_assets/entity_extraction/validated_output_skeleton_reask_1.py b/tests/integration_tests/test_assets/entity_extraction/validated_output_skeleton_reask_1.py
@@ -1,4 +1,4 @@
-# flake8: noqa: E501
+# ruff: noqa: E501
 from guardrails.utils.reask_utils import SkeletonReAsk
 from guardrails.validators import FailResult
 
diff --git a/tests/integration_tests/test_assets/entity_extraction/validated_output_skeleton_reask_2.py b/tests/integration_tests/test_assets/entity_extraction/validated_output_skeleton_reask_2.py
@@ -1,4 +1,4 @@
-# flake8: noqa: E501
+# ruff: noqa: E501
 VALIDATED_OUTPUT_SKELETON_REASK_2 = {
     "fees": [
         {"name": "annual membership", "explanation": "", "value": 0.0},
diff --git a/tests/integration_tests/test_assets/pydantic/__init__.py b/tests/integration_tests/test_assets/pydantic/__init__.py
@@ -1,4 +1,4 @@
-# flake8: noqa: E501
+# ruff: noqa: E501
 import os
 
 from .msg_validated_output_reask import MSG_VALIDATED_OUTPUT_REASK
diff --git a/tests/integration_tests/test_assets/pydantic/validated_response_reask.py b/tests/integration_tests/test_assets/pydantic/validated_response_reask.py
@@ -1,4 +1,4 @@
-# flake8: noqa: E501
+# ruff: noqa: E501
 from typing import Any, Dict, List
 
 from pydantic import BaseModel, Field
diff --git a/tests/integration_tests/test_assets/python_rail/__init__.py b/tests/integration_tests/test_assets/python_rail/__init__.py
@@ -1,4 +1,4 @@
-# flake8: noqa: E501
+# ruff: noqa: E501
 import os
 
 DATA_DIR = os.path.join(os.path.abspath(os.path.dirname(__file__)))
@@ -26,11 +26,9 @@
 RAIL_SPEC_WITH_VALIDATOR_PARALLELISM = reader("validator_parallelism.rail")
 VALIDATOR_PARALLELISM_PROMPT_1 = reader("validator_parallelism_prompt_1.txt")
 VALIDATOR_PARALLELISM_RESPONSE_1 = reader("validator_parallelism_1.txt")
-from .validator_parallelism_reask_1 import VALIDATOR_PARALLELISM_REASK_1
 
 VALIDATOR_PARALLELISM_PROMPT_2 = reader("validator_parallelism_prompt_2.txt")
 VALIDATOR_PARALLELISM_RESPONSE_2 = reader("validator_parallelism_2.txt")
-from .validator_parallelism_reask_2 import VALIDATOR_PARALLELISM_REASK_2
 
 VALIDATOR_PARALLELISM_PROMPT_3 = reader("validator_parallelism_prompt_3.txt")
 VALIDATOR_PARALLELISM_RESPONSE_3 = reader("validator_parallelism_3.txt")
diff --git a/tests/integration_tests/test_assets/string/__init__.py b/tests/integration_tests/test_assets/string/__init__.py
@@ -1,4 +1,4 @@
-# flake8: noqa: E501
+# ruff: noqa: E501
 import os
 
 from .msg_validated_output_reask import MSG_VALIDATED_OUTPUT_REASK
diff --git a/tests/integration_tests/test_cli.py b/tests/integration_tests/test_cli.py
@@ -1,4 +1,4 @@
-# flake8: noqa: E501
+# ruff: noqa: E501
 
 import json
 import os
diff --git a/tests/integration_tests/test_data_validation.py b/tests/integration_tests/test_data_validation.py
@@ -1,5 +1,5 @@
-# flake8: noqa: E501
-from typing import Literal, Optional, Union
+# ruff: noqa: E501
+from typing import Literal, Union
 
 import pytest
 from pydantic import BaseModel, Field
diff --git a/tests/integration_tests/test_schema_to_prompt.py b/tests/integration_tests/test_schema_to_prompt.py
@@ -1,5 +1,4 @@
-# flake8: noqa: E501
-import pytest
+# ruff: noqa: E501
 
 from guardrails import Guard
 

Original file line number	Diff line number	Diff line change
`@@ -1,4 +1,4 @@`
`1`		`-# flake8: noqa: E501`
	`1`	`+# ruff: noqa: E501`
`2`	`2`	`import os`
`3`	`3`
`4`	`4`	`from .optional_prompts import (`