|
15 | 15 | Hashable, |
16 | 16 | Iterable, |
17 | 17 | List, |
| 18 | + Literal, |
18 | 19 | Mapping, |
19 | 20 | Sequence, |
20 | 21 | Tuple, |
@@ -716,7 +717,10 @@ def _infer_types(self, values, na_values, cast_type, try_num_bool: bool = True): |
716 | 717 | np.putmask(values, mask, np.nan) |
717 | 718 | return values, na_count |
718 | 719 |
|
719 | | - use_nullable_dtypes = self.use_nullable_dtypes and cast_type is None |
| 720 | + use_nullable_dtypes: Literal[True] | Literal[False] = ( |
| 721 | + self.use_nullable_dtypes and cast_type is None |
| 722 | + ) |
| 723 | + result: ArrayLike |
720 | 724 |
|
721 | 725 | if try_num_bool and is_object_dtype(values.dtype): |
722 | 726 | # exclude e.g DatetimeIndex here |
@@ -753,16 +757,16 @@ def _infer_types(self, values, na_values, cast_type, try_num_bool: bool = True): |
753 | 757 | na_count = parsers.sanitize_objects(values, na_values) |
754 | 758 |
|
755 | 759 | if result.dtype == np.object_ and try_num_bool: |
756 | | - result, mask = libops.maybe_convert_bool( |
| 760 | + result, bool_mask = libops.maybe_convert_bool( |
757 | 761 | np.asarray(values), |
758 | 762 | true_values=self.true_values, |
759 | 763 | false_values=self.false_values, |
760 | 764 | convert_to_masked_nullable=use_nullable_dtypes, |
761 | 765 | ) |
762 | 766 | if result.dtype == np.bool_ and use_nullable_dtypes: |
763 | | - if mask is None: |
764 | | - mask = np.zeros(result.shape, dtype=np.bool_) |
765 | | - result = BooleanArray(result, mask) |
| 767 | + if bool_mask is None: |
| 768 | + bool_mask = np.zeros(result.shape, dtype=np.bool_) |
| 769 | + result = BooleanArray(result, bool_mask) |
766 | 770 | elif result.dtype == np.object_ and use_nullable_dtypes: |
767 | 771 | result = StringDtype().construct_array_type()._from_sequence(values) |
768 | 772 |
|
|
0 commit comments