fix: PLC0415

Author: kukushking

awswrangler/_distributed.py

@@ -117,7 +117,7 @@ def register(cls, name: EngineLiteral | None = None) -> None:
             cls._registry.clear()
 
             if engine_name == EngineEnum.RAY.value:
-                from awswrangler.distributed.ray._register import register_ray
+                from awswrangler.distributed.ray._register import register_ray  # noqa: PLC0415
 
                 register_ray()
 
@@ -127,7 +127,7 @@ def initialize(cls, name: EngineLiteral | None = None) -> None:
         with cls._lock:
             engine_name = name or cls.get_installed().value
             if engine_name == EngineEnum.RAY.value:
-                from awswrangler.distributed.ray import initialize_ray
+                from awswrangler.distributed.ray import initialize_ray  # noqa: PLC0415
 
                 initialize_ray()
             cls._initialized_engine = EngineEnum[engine_name.upper()]
@@ -187,7 +187,7 @@ def set(cls, name: EngineLiteral) -> None:
 
 def _reload() -> None:
     """Reload Pandas proxy module."""
-    import awswrangler.pandas
+    import awswrangler.pandas  # noqa: PLC0415
 
     reload(awswrangler.pandas)
 

awswrangler/distributed/ray/_register.py

@@ -1,4 +1,5 @@
 """Ray and Modin registered methods (PRIVATE)."""
+# ruff: noqa: PLC0415
 
 from awswrangler._data_types import pyarrow_types_from_pandas
 from awswrangler._distributed import MemoryFormatEnum, engine, memory_format

awswrangler/distributed/ray/datasources/arrow_csv_datasource.py

@@ -1,4 +1,5 @@
 """Ray ArrowCSVDatasource Module."""
+# ruff: noqa: PLC0415
 
 from __future__ import annotations
 

awswrangler/distributed/ray/datasources/arrow_orc_datasink.py

@@ -1,4 +1,5 @@
 """Ray PandasTextDatasink Module."""
+# ruff: noqa: PLC0415
 
 from __future__ import annotations
 

awswrangler/distributed/ray/datasources/arrow_orc_datasource.py

@@ -1,4 +1,5 @@
 """Ray ArrowCSVDatasource Module."""
+# ruff: noqa: PLC0415
 
 from __future__ import annotations
 

awswrangler/distributed/ray/datasources/arrow_parquet_datasource.py

@@ -5,6 +5,7 @@
 and customized to ensure compatibility with AWS SDK for pandas behavior. Changes from the original implementation,
 are documented in the comments and marked with (AWS SDK for pandas) prefix.
 """
+# ruff: noqa: PLC0415
 
 from __future__ import annotations
 

awswrangler/opensearch/_write.py

@@ -117,7 +117,7 @@ def _file_line_generator(path: str, is_json: bool = False) -> Generator[Any, Non
 
 @_utils.check_optional_dependency(jsonpath_ng, "jsonpath_ng")
 def _get_documents_w_json_path(documents: list[Mapping[str, Any]], json_path: str) -> list[Any]:
-    from jsonpath_ng.exceptions import JsonPathParserError
+    from jsonpath_ng.exceptions import JsonPathParserError  # noqa: PLC0415
 
     try:
         jsonpath_expression = jsonpath_ng.parse(json_path)

awswrangler/s3/_read_orc.py

@@ -47,7 +47,7 @@
 
 
 def _pyarrow_orc_file_wrapper(source: Any) -> "ORCFile":
-    from pyarrow.orc import ORCFile
+    from pyarrow.orc import ORCFile  # noqa: PLC0415
 
     try:
         return ORCFile(source=source)

awswrangler/s3/_write_orc.py

@@ -57,7 +57,7 @@ def _new_writer(
     s3_additional_kwargs: dict[str, str] | None,
     use_threads: bool | int,
 ) -> Iterator["ORCWriter"]:
-    from pyarrow.orc import ORCWriter
+    from pyarrow.orc import ORCWriter  # noqa: PLC0415
 
     writer: "ORCWriter" | None = None
     if not pyarrow_additional_kwargs: