Skip to content

Commit b956f9f

Browse files
committed
Share declaration
1 parent 5495551 commit b956f9f

File tree

3 files changed

+5
-6
lines changed

3 files changed

+5
-6
lines changed

pandas/_libs/parsers.pyx

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1331,8 +1331,8 @@ cdef class TextReader:
13311331
return None
13321332

13331333

1334-
cdef object _true_values = [b'True', b'TRUE', b'true']
1335-
cdef object _false_values = [b'False', b'FALSE', b'false']
1334+
_true_values = [b'True', b'TRUE', b'true']
1335+
_false_values = [b'False', b'FALSE', b'false']
13361336

13371337

13381338
def _ensure_encoded(list lst):

pandas/core/arrays/categorical.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -510,6 +510,7 @@ def _from_inferred_categories(cls, inferred_categories, inferred_codes,
510510
Categorical
511511
"""
512512
from pandas import Index, to_numeric, to_datetime, to_timedelta
513+
from pandas._libs.parsers import _true_values
513514

514515
cats = Index(inferred_categories)
515516

@@ -525,10 +526,8 @@ def _from_inferred_categories(cls, inferred_categories, inferred_codes,
525526
elif is_timedelta64_dtype(dtype.categories):
526527
cats = to_timedelta(inferred_categories, errors='coerce')
527528
elif dtype.categories.is_boolean():
528-
# from _true_values in parsers.pyx
529-
# TODO: share declaration with parsers.pyx
530529
if true_values is None:
531-
true_values = [b'True', b'true', b'TRUE']
530+
true_values = _true_values
532531
true_values = [x.decode() for x in true_values]
533532
cats = cats.isin(true_values)
534533

pandas/tests/io/parser/dtypes.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -236,7 +236,7 @@ def test_categoricaldtype_coerces_timedelta(self):
236236
'b\nTrue\nFalse\nNA\nFALSE',
237237
])
238238
def test_categoricaldtype_coerces_boolen(self, data):
239-
# 20498
239+
# GH-20498
240240
dtype = {'b': CategoricalDtype([False, True])}
241241
expected = pd.DataFrame({"b": Categorical([True, False, None, False])})
242242
result = self.read_csv(StringIO(data), dtype=dtype)

0 commit comments

Comments
 (0)