fix: ruff errors

pandasai/core/response/__init__.py

@@ -1,9 +1,9 @@
-from .parser import ResponseParser
 from .base import BaseResponse
-from .string import StringResponse
-from .number import NumberResponse
-from .dataframe import DataFrameResponse
 from .chart import ChartResponse
+from .dataframe import DataFrameResponse
+from .number import NumberResponse
+from .parser import ResponseParser
+from .string import StringResponse
 
 __all__ = [
     "ResponseParser",

pandasai/core/response/base.py

@@ -1,5 +1,5 @@
-from typing import Any
 import json
+from typing import Any
 
 
 class BaseResponse:

pandasai/core/response/chart.py

@@ -1,7 +1,8 @@
-from typing import Any
-from PIL import Image
 import base64
 import io
+from typing import Any
+
+from PIL import Image
 
 from .base import BaseResponse
 

pandasai/core/response/dataframe.py

@@ -1,4 +1,5 @@
 from typing import Any
+
 import pandas as pd
 
 from .base import BaseResponse

pandasai/core/response/parser.py

@@ -6,10 +6,10 @@
 from pandasai.exceptions import InvalidOutputValueMismatch
 
 from .base import BaseResponse
+from .chart import ChartResponse
+from .dataframe import DataFrameResponse
 from .number import NumberResponse
 from .string import StringResponse
-from .dataframe import DataFrameResponse
-from .chart import ChartResponse
 
 
 class ResponseParser:

pandasai/dataframe/base.py

@@ -12,14 +12,14 @@
 
 import pandasai as pai
 from pandasai.config import Config
+from pandasai.core.response import BaseResponse
 from pandasai.exceptions import DatasetNotFound, PandasAIApiKeyError
 from pandasai.helpers.dataframe_serializer import (
     DataframeSerializer,
     DataframeSerializerType,
 )
 from pandasai.helpers.path import find_project_root
 from pandasai.helpers.request import get_pandaai_session
-from pandasai.core.response import BaseResponse
 
 if TYPE_CHECKING:
     from pandasai.agent.base import Agent

tests/unit_tests/helpers/test_responses.py

@@ -2,13 +2,13 @@
 
 import pandas as pd
 
-from pandasai.core.response.parser import ResponseParser
 from pandasai.core.response import (
     ChartResponse,
     DataFrameResponse,
     NumberResponse,
     StringResponse,
 )
+from pandasai.core.response.parser import ResponseParser
 from pandasai.exceptions import InvalidOutputValueMismatch
 
 

tests/unit_tests/response/test_chart_response.py

@@ -1,7 +1,9 @@
-import pytest
-from PIL import Image
 import base64
 import io
+
+import pytest
+from PIL import Image
+
 from pandasai.core.response.chart import ChartResponse
 
 

tests/unit_tests/response/test_dataframe_response.py

@@ -1,5 +1,6 @@
-import pytest
 import pandas as pd
+import pytest
+
 from pandasai.core.response.dataframe import DataFrameResponse