fix linter

Author: zhengruifeng

python/pyspark/sql/tests/arrow/test_arrow_cogrouped_map.py

@@ -335,6 +335,8 @@ def summarize(key, left, right):
                 }
             )
 
+        schema = "key long, left_rows long, left_columns long, right_rows long, right_columns long"
+
         expected = [
             Row(key=0, left_rows=5000000, left_columns=12, right_rows=25000, right_columns=22),
             Row(key=1, left_rows=5000000, left_columns=12, right_rows=25000, right_columns=22),
@@ -353,10 +355,7 @@ def summarize(key, left, right):
                     result = (
                         df1.groupby("key")
                         .cogroup(df2.groupby("key"))
-                        .applyInArrow(
-                            summarize,
-                            schema="key long, left_rows long, left_columns long, right_rows long, right_columns long",
-                        )
+                        .applyInArrow(summarize, schema=schema)
                         .sort("key")
                         .collect()
                     )

python/pyspark/sql/tests/pandas/test_pandas_cogrouped_map.py

@@ -682,6 +682,8 @@ def summarize(key, left, right):
                 }
             )
 
+        schema = "key long, left_rows long, left_columns long, right_rows long, right_columns long"
+
         expected = [
             Row(key=0, left_rows=5000000, left_columns=12, right_rows=25000, right_columns=22),
             Row(key=1, left_rows=5000000, left_columns=12, right_rows=25000, right_columns=22),
@@ -700,10 +702,7 @@ def summarize(key, left, right):
                     result = (
                         df1.groupby("key")
                         .cogroup(df2.groupby("key"))
-                        .applyInPandas(
-                            summarize,
-                            schema="key long, left_rows long, left_columns long, right_rows long, right_columns long",
-                        )
+                        .applyInPandas(summarize, schema=schema)
                         .sort("key")
                         .collect()
                     )