From 9dd44c6c3f9988dc45e9777fedae5c3a6d1ecd18 Mon Sep 17 00:00:00 2001 From: Anthony Mahanna Date: Fri, 21 Jul 2023 00:08:37 -0400 Subject: [PATCH] fix: mypy --- tests/test_adapter.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_adapter.py b/tests/test_adapter.py index ead5bf3..c959177 100644 --- a/tests/test_adapter.py +++ b/tests/test_adapter.py @@ -1003,7 +1003,7 @@ def assert_pyg_to_adb_meta( assert df[meta_val].values.tolist() == data.tolist() if callable(meta_val): - udf_df = meta_val(data) + udf_df = meta_val(data, DataFrame(index=range(len(data)))) assert all([column in df for column in udf_df.columns]) for column in udf_df.columns: assert df[column].tolist() == udf_df[column].tolist()