Skip to content

Commit

Permalink
fix lint
Browse files Browse the repository at this point in the history
  • Loading branch information
bjchambers committed Oct 16, 2023
1 parent bbbe6bb commit 688b51a
Showing 1 changed file with 4 additions and 1 deletion.
5 changes: 4 additions & 1 deletion python/pytests/execution_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -41,17 +41,20 @@ async def test_iter_rows(source_int64) -> None:
with pytest.raises(StopIteration):
next(results)


@kd.udf("add_one<N: number>(x: N) -> N")
def add_one(x: pd.Series) -> pd.Series:
"""Use Pandas to one"""
print("In UDF")
return (x + 1)
return x + 1


async def test_iter_udf(source_int64) -> None:
results = source_int64.col("m").pipe(add_one).run_iter("row", row_limit=2)
assert next(results)["result"] == 6
assert next(results)["result"] == 25


async def test_iter_pandas_async(golden, source_int64) -> None:
batches: AsyncIterator[pd.DataFrame] = source_int64.run_iter(
row_limit=4, max_batch_size=2
Expand Down

0 comments on commit 688b51a

Please sign in to comment.