Skip to content

Commit

Permalink
Skip deleting copied dask dataframe to avoid crash while creating fro…
Browse files Browse the repository at this point in the history
…m cudf edge lists
  • Loading branch information
naimnv committed Sep 22, 2023
1 parent 0d8ff97 commit 9bb51fe
Showing 1 changed file with 1 addition and 3 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -1193,7 +1193,5 @@ def _get_column_from_ls_dfs(lst_df, col_name):
if len_df == 0:
return lst_df[0][col_name]
output_col = cudf.concat([df[col_name] for df in lst_df], ignore_index=True)
for df in lst_df:
df.drop(columns=[col_name], inplace=True)
gc.collect()
# FIXME: For now, don't delete the copied dataframe to avoid crash
return output_col

0 comments on commit 9bb51fe

Please sign in to comment.