Skip to content

Commit 045da54

Browse files
authored
Merge pull request #153 from Roestlab/feature/polars_explode
change explode to using polars
2 parents ba7495d + b0a01bb commit 045da54

1 file changed

Lines changed: 3 additions & 1 deletion

File tree

pyprophet/io/_base.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -384,7 +384,9 @@ def _prepare_score_dataframe(
384384
# logger.info(
385385
# "Exploding protein_id column to handle multiple proteins per feature."
386386
# )
387-
df = df.explode("protein_id")
387+
pl_df = pl.from_pandas(df)
388+
pl_df = pl_df.explode("protein_id")
389+
df = pl_df.to_pandas()
388390

389391
if level == "transition":
390392
if self.file_type in ("parquet", "parquet_split", "parquet_split_multi"):

0 commit comments

Comments
 (0)