Skip to content

Commit 6bb9ab8

Browse files
ryankarlosjreback
authored andcommitted
Remove pandas.io.stata from Typing Blacklist (#25940)
1 parent 2769ebf commit 6bb9ab8

File tree

2 files changed

+6
-9
lines changed

2 files changed

+6
-9
lines changed

mypy.ini

-3
Original file line numberDiff line numberDiff line change
@@ -164,9 +164,6 @@ ignore_errors=True
164164
[mypy-pandas.io.pytables]
165165
ignore_errors=True
166166

167-
[mypy-pandas.io.stata]
168-
ignore_errors=True
169-
170167
[mypy-pandas.util._doctools]
171168
ignore_errors=True
172169

pandas/io/stata.py

+6-6
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,6 @@
2222
import numpy as np
2323

2424
from pandas._libs.lib import infer_dtype
25-
from pandas._libs.tslibs import NaT, Timestamp
2625
from pandas._libs.writers import max_len_string_array
2726
from pandas.compat import lmap, lrange, lzip
2827
from pandas.util._decorators import Appender, deprecate_kwarg
@@ -31,8 +30,8 @@
3130
ensure_object, is_categorical_dtype, is_datetime64_dtype)
3231

3332
from pandas import (
34-
DatetimeIndex, compat, concat, isna, to_datetime, to_timedelta)
35-
from pandas.core.arrays import Categorical
33+
Categorical, DatetimeIndex, NaT, Timestamp, compat, concat, isna,
34+
to_datetime, to_timedelta)
3635
from pandas.core.base import StringMixin
3736
from pandas.core.frame import DataFrame
3837
from pandas.core.series import Series
@@ -1050,7 +1049,7 @@ def _read_new_header(self, first_char):
10501049

10511050
self.nobs = self._get_nobs()
10521051
self.path_or_buf.read(11) # </N><label>
1053-
self.data_label = self._get_data_label()
1052+
self._data_label = self._get_data_label()
10541053
self.path_or_buf.read(19) # </label><timestamp>
10551054
self.time_stamp = self._get_time_stamp()
10561055
self.path_or_buf.read(26) # </timestamp></header><map>
@@ -1237,7 +1236,7 @@ def _read_old_header(self, first_char):
12371236
self.path_or_buf.read(2))[0]
12381237
self.nobs = self._get_nobs()
12391238

1240-
self.data_label = self._get_data_label()
1239+
self._data_label = self._get_data_label()
12411240

12421241
self.time_stamp = self._get_time_stamp()
12431242

@@ -1740,11 +1739,12 @@ def _do_convert_categoricals(self, data, value_label_dict, lbllist,
17401739
data = DataFrame.from_dict(OrderedDict(cat_converted_data))
17411740
return data
17421741

1742+
@property
17431743
def data_label(self):
17441744
"""
17451745
Return data label of Stata file.
17461746
"""
1747-
return self.data_label
1747+
return self._data_label
17481748

17491749
def variable_labels(self):
17501750
"""

0 commit comments

Comments
 (0)