Skip to content
This repository was archived by the owner on Feb 2, 2024. It is now read-only.

Commit d59346e

Browse files
author
Ehsan Totoni
committed
fix split view bool index refcount
1 parent 3918e54 commit d59346e

2 files changed

Lines changed: 4 additions & 1 deletion

File tree

hpat/hiframes/split_impl.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -278,6 +278,8 @@ def codegen(context, builder, sig, args):
278278
out_view.data_offsets = view_payload.data_offsets
279279
# TODO: incref?
280280
out_view.data = data_ptr
281+
if context.enable_nrt:
282+
context.nrt.incref(builder, data_t, data_ptr)
281283
# out_view.null_bitmap = view_payload.null_bitmap
282284
out_view.meminfo = meminfo
283285
ret = out_view._getvalue()

hpat/tests/test_hiframes.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -381,7 +381,8 @@ def test_impl(df):
381381

382382
df = pd.DataFrame({'A': ['AB,CC', 'C,ABB,D', 'G', '', 'g,f']})
383383
hpat_func = hpat.jit(test_impl)
384-
pd.testing.assert_frame_equal(hpat_func(df), test_impl(df))
384+
pd.testing.assert_frame_equal(
385+
hpat_func(df), test_impl(df).reset_index(drop=True))
385386

386387
def test_str_split_box_df(self):
387388
def test_impl(df):

0 commit comments

Comments
 (0)