Applying black style. #338

Author: igorborgest

awswrangler/catalog.py

@@ -1342,7 +1342,7 @@ def _csv_table_definition(
     partitions_types: Dict[str, str],
     compression: Optional[str],
     sep: str,
-    skip_header_line_count: Optional[int]
+    skip_header_line_count: Optional[int],
 ) -> Dict[str, Any]:
     compressed: bool = compression is not None
     parameters: Dict[str, str] = {

tests/test_athena_csv.py

@@ -327,21 +327,16 @@ def test_skip_header(path, glue_database, glue_table, use_threads, ctas_approach
     df = pd.DataFrame({"c0": [1, 2], "c1": [3.3, 4.4], "c2": ["foo", "boo"]})
     df["c0"] = df["c0"].astype("Int64")
     df["c2"] = df["c2"].astype("string")
-    paths = wr.s3.to_csv(
-        df=df,
-        path=f"{path}0.csv",
-        sep=",",
-        index=False,
-        header=True,
-        use_threads=use_threads
-    )["paths"]
+    paths = wr.s3.to_csv(df=df, path=f"{path}0.csv", sep=",", index=False, header=True, use_threads=use_threads)[
+        "paths"
+    ]
     wr.s3.wait_objects_exist(paths=paths, use_threads=use_threads)
     wr.catalog.create_csv_table(
         database=glue_database,
         table=glue_table,
         path=path,
         columns_types={"c0": "bigint", "c1": "double", "c2": "string"},
-        skip_header_line_count=1
+        skip_header_line_count=1,
     )
     df2 = wr.athena.read_sql_table(glue_table, glue_database, use_threads=use_threads, ctas_approach=ctas_approach)
     assert df.equals(df2)