diff --git a/pybench/utils.py b/pybench/utils.py index 53eeef8..b665455 100644 --- a/pybench/utils.py +++ b/pybench/utils.py @@ -197,7 +197,7 @@ def split_params_list(df, params_name, columns=None): >>> ["params.shape.0", "params.shape.1"]) >>> split_params_list(df, "params.shape", ["shape0", "shape1"]) """ - lst = df[params_name].to_list() + lst = df[params_name].tolist() lst = [[l] if not isinstance(l, list) else l for l in lst] ncols = max([len(l) for l in lst]) if columns is None: