fix: Convert Polars container dtype entries to list in _get_cell (#778) #794
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
Summary
This pull request adds a small fix that ensures that
_get_cellreturns alistrather thanpl.Seriesif cell entry is a container dtype (pl.Listorpl.Array). This makes its behavior with polars dataframes consistent with _get_cell for pandas data frames which return lists for container dtypes.I went ahead and thought this was the best place to address #778 which surfaced as a syntax bug when running one of the
gt-extrastutorials with polars. Happy to consider other solutions in case it warrants a fix elsewhere in the code base!Changes
_tbl_data._get_cell: check if retrieved cell entry is either a container data type and convert to list.tests/test_tbl_data.py: addsdf_container_dtypesfixture andtest_get_cell_container_dtypestestRelated GitHub Issues and PRs
PlDataFrame._get_cell()type inconsistency breaks gt-extrasgt_plt_bar_stack#778Checklist