From e8cacd6a79c7cd654c280cc83d208a84ac18b558 Mon Sep 17 00:00:00 2001 From: julian fong Date: Wed, 14 Aug 2024 21:27:08 -0400 Subject: [PATCH] refactored lazy frames to use .collect_schema().names() to fix warning --- skpro/datatypes/_adapter/polars.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/skpro/datatypes/_adapter/polars.py b/skpro/datatypes/_adapter/polars.py index ca075e9c..647d41f5 100644 --- a/skpro/datatypes/_adapter/polars.py +++ b/skpro/datatypes/_adapter/polars.py @@ -34,7 +34,10 @@ def check_polars_frame(obj, return_metadata=False, var_name="obj", lazy=False): if _req("n_features", return_metadata): metadata["n_features"] = obj.width if _req("feature_names", return_metadata): - metadata["feature_names"] = obj.columns + if lazy: + metadata["feature_names"] = obj.collect_schema().names() + else: + metadata["feature_names"] = obj.columns # check if there are any nans # compute only if needed