Skip to content

Commit 1802cbd

Browse files
committed
formatting fixes
1 parent 0ecba8a commit 1802cbd

File tree

1 file changed

+5
-4
lines changed

1 file changed

+5
-4
lines changed

examples/src/main/python/sql/arrow.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -257,19 +257,20 @@ def filter_func(batch_iter):
257257
# +---+---+
258258
# $example off:map_iter_pandas_udf$
259259

260+
260261
def cogrouped_map_pandas_udf_example(spark):
261262
# $example on:cogrouped_map_pandas_udf$
262263
import pandas as pd
263264

264265
from pyspark.sql.functions import pandas_udf, PandasUDFType
265266

266267
df1 = spark.createDataFrame(
267-
[(20000101, 1, 1.0), (20000101, 2, 2.0), (20000102, 1, 3.0), (20000102, 2, 4.0)],
268-
("time", "id", "v1"))
268+
[(20000101, 1, 1.0), (20000101, 2, 2.0), (20000102, 1, 3.0), (20000102, 2, 4.0)],
269+
("time", "id", "v1"))
269270

270271
df2 = spark.createDataFrame(
271-
[(20000101, 1, "x"), (20000101, 2, "y")],
272-
("time", "id", "v2"))
272+
[(20000101, 1, "x"), (20000101, 2, "y")],
273+
("time", "id", "v2"))
273274

274275
@pandas_udf("time int, id int, v1 double, v2 string", PandasUDFType.COGROUPED_MAP)
275276
def asof_join(l, r):

0 commit comments

Comments
 (0)