fix: types

Author: kukushking

awswrangler/_distributed.py

@@ -112,7 +112,7 @@ def wrapper(*args: Any, **kw: dict[str, Any]) -> Any:
     def register(cls, name: EngineLiteral | None = None) -> None:
         """Register the distribution engine dispatch methods."""
         with cls._lock:
-            engine_name = cast(EngineLiteral, name or cls.get().value)
+            engine_name = name or cls.get().value
             cls.set(engine_name)
             cls._registry.clear()
 
@@ -125,7 +125,7 @@ def register(cls, name: EngineLiteral | None = None) -> None:
     def initialize(cls, name: EngineLiteral | None = None) -> None:
         """Initialize the distribution engine."""
         with cls._lock:
-            engine_name = cast(EngineLiteral, name or cls.get_installed().value)
+            engine_name = name or cls.get_installed().value
             if engine_name == EngineEnum.RAY.value:
                 from awswrangler.distributed.ray import initialize_ray
 
@@ -136,7 +136,7 @@ def initialize(cls, name: EngineLiteral | None = None) -> None:
     def is_initialized(cls, name: EngineLiteral | None = None) -> bool:
         """Check if the distribution engine is initialized."""
         with cls._lock:
-            engine_name = cast(EngineLiteral, name or cls.get_installed().value)
+            engine_name = name or cls.get_installed().value
 
             return False if not cls._initialized_engine else cls._initialized_engine.value == engine_name
 

awswrangler/athena/_write_iceberg.py

@@ -501,10 +501,7 @@ def to_iceberg(  # noqa: PLR0913
         merge_condition=merge_condition,
     )
 
-    glue_table_settings = cast(
-        GlueTableSettings,
-        glue_table_settings if glue_table_settings else {},
-    )
+    glue_table_settings = glue_table_settings if glue_table_settings else {}
 
     try:
         # Create Iceberg table if it doesn't exist

awswrangler/catalog/_utils.py

@@ -196,7 +196,7 @@ def sanitize_dataframe_columns_names(df: pd.DataFrame, handle_duplicate_columns:
     """
     df.columns = [sanitize_column_name(x) for x in df.columns]
     df.index.names = [None if x is None else sanitize_column_name(x) for x in df.index.names]
-    if df.columns.duplicated().any():  # type: ignore[attr-defined]
+    if df.columns.duplicated().any():
         if handle_duplicate_columns == "warn":
             warnings.warn(
                 "Duplicate columns were detected, consider using `handle_duplicate_columns='[drop|rename]'`",

awswrangler/quicksight/_get_list.py

@@ -28,7 +28,7 @@ def _list(
     client = _utils.client(service_name="quicksight", session=boto3_session)
     func: Callable[..., dict[str, Any]] = getattr(client, func_name)
     response: dict[str, Any] = func(AwsAccountId=account_id, **kwargs)
-    next_token: str = response.get("NextToken", None)
+    next_token: str | None = response.get("NextToken", None)
     result: list[dict[str, Any]] = response[attr_name]
     while next_token is not None:
         response = func(AwsAccountId=account_id, NextToken=next_token, **kwargs)

awswrangler/s3/_write_orc.py

@@ -620,10 +620,7 @@ def to_orc(
     }
 
     """
-    glue_table_settings = cast(
-        GlueTableSettings,
-        glue_table_settings if glue_table_settings else {},
-    )
+    glue_table_settings = glue_table_settings if glue_table_settings else {}
 
     table_type = glue_table_settings.get("table_type")
     description = glue_table_settings.get("description")

awswrangler/s3/_write_parquet.py

@@ -678,10 +678,7 @@ def to_parquet(
     }
 
     """
-    glue_table_settings = cast(
-        GlueTableSettings,
-        glue_table_settings if glue_table_settings else {},
-    )
+    glue_table_settings = glue_table_settings if glue_table_settings else {}
 
     table_type = glue_table_settings.get("table_type")
     description = glue_table_settings.get("description")

awswrangler/s3/_write_text.py

@@ -435,10 +435,7 @@ def to_csv(  # noqa: PLR0912,PLR0915
             "e.g. wr.s3.to_csv(df, path, sep='|', na_rep='NULL', decimal=',', compression='gzip')"
         )
 
-    glue_table_settings = cast(
-        GlueTableSettings,
-        glue_table_settings if glue_table_settings else {},
-    )
+    glue_table_settings = glue_table_settings if glue_table_settings else {}
 
     table_type = glue_table_settings.get("table_type")
     description = glue_table_settings.get("description")
@@ -885,10 +882,7 @@ def to_json(  # noqa: PLR0912,PLR0915
             "e.g. wr.s3.to_json(df, path, lines=True, date_format='iso')"
         )
 
-    glue_table_settings = cast(
-        GlueTableSettings,
-        glue_table_settings if glue_table_settings else {},
-    )
+    glue_table_settings = glue_table_settings if glue_table_settings else {}
 
     table_type = glue_table_settings.get("table_type")
     description = glue_table_settings.get("description")