From 446e180d477aa09d2676539346f82eed68d7ef19 Mon Sep 17 00:00:00 2001 From: julian fong Date: Wed, 14 Aug 2024 17:56:08 -0400 Subject: [PATCH] fixed name to only include original index name in returned dataframe --- skpro/datatypes/_adapter/polars.py | 1 + 1 file changed, 1 insertion(+) diff --git a/skpro/datatypes/_adapter/polars.py b/skpro/datatypes/_adapter/polars.py index 1f337a5d..4892c800 100644 --- a/skpro/datatypes/_adapter/polars.py +++ b/skpro/datatypes/_adapter/polars.py @@ -72,6 +72,7 @@ def convert_polars_to_pandas_with_index(obj): for col in obj.columns: if col.startswith("__index__"): pd_df = pd_df.set_index(col, drop=True) + pd_df.index.name = col.split("__index__")[1] return pd_df