Skip to content

Commit 7d98727

Browse files
fix failing test_factorize_empty
1 parent ea59c38 commit 7d98727

File tree

1 file changed

+7
-1
lines changed

1 file changed

+7
-1
lines changed

pandas/core/arrays/string_arrow.py

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -287,7 +287,13 @@ def factorize(self, na_sentinel: int = -1) -> Tuple[np.ndarray, ExtensionArray]:
287287
indices = indices.astype(int)
288288
if not is_int64_dtype(indices):
289289
indices = indices.astype(np.int64)
290-
return indices.values, type(self)(encoded.chunk(0).dictionary)
290+
291+
if encoded.num_chunks:
292+
uniques = type(self)(encoded.chunk(0).dictionary)
293+
else:
294+
uniques = type(self)(pa.array([], type=encoded.type.value_type))
295+
296+
return indices.values, uniques
291297

292298
@classmethod
293299
def _concat_same_type(cls, to_concat) -> ArrowStringArray:

0 commit comments

Comments
 (0)