Skip to content

Commit e37a96a

Browse files
committed
revert changes for integerarray & booleanarray
1 parent a7f708b commit e37a96a

File tree

1 file changed

+2
-6
lines changed

1 file changed

+2
-6
lines changed

connectorx-python/connectorx/__init__.py

Lines changed: 2 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -459,20 +459,16 @@ def reconstruct_pandas(df_infos: _DataframeInfos) -> pd.DataFrame:
459459
pd.core.internals.make_block(block_data, placement=binfo.cids)
460460
)
461461
elif binfo.dt == 1: # IntegerArray
462-
integer_array = pd.core.arrays.IntegerArray._from_sequence(block_data[0])
463-
integer_array._mask = block_data[1]
464462
blocks.append(
465463
pd.core.internals.make_block(
466-
integer_array,
464+
pd.core.arrays.IntegerArray(block_data[0], block_data[1]),
467465
placement=binfo.cids[0],
468466
)
469467
)
470468
elif binfo.dt == 2: # BooleanArray
471-
bool_array = pd.core.arrays.BooleanArray._from_sequence(block_data[0])
472-
bool_array._mask = block_data[1]
473469
blocks.append(
474470
pd.core.internals.make_block(
475-
bool_array,
471+
pd.core.arrays.BooleanArray(block_data[0], block_data[1]),
476472
placement=binfo.cids[0],
477473
)
478474
)

0 commit comments

Comments
 (0)