fix: mypy

arangoml · Jul 21, 2023 · 9dd44c6 · 9dd44c6
1 parent 2a6abad
commit 9dd44c6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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()