diff --git a/benchmarks/benchmarks.py b/benchmarks/benchmarks.py index 26fc7be..a220f98 100644 --- a/benchmarks/benchmarks.py +++ b/benchmarks/benchmarks.py @@ -31,8 +31,8 @@ def _generate_benchmark_data(add_nested=True): layer_nf = npd.NestedFrame(data=layer_data).set_index("index").sort_index() # Convert to Dask - base_nf = nd.NestedFrame.from_nested_pandas(base_nf).repartition(npartitions=5) - layer_nf = nd.NestedFrame.from_nested_pandas(layer_nf).repartition(npartitions=50) + base_nf = nd.NestedFrame.from_pandas(base_nf).repartition(npartitions=5) + layer_nf = nd.NestedFrame.from_pandas(layer_nf).repartition(npartitions=50) # Return based on add_nested if add_nested: