Skip to content

Commit da82aef

Browse files
authored
de-privatize (#36259)
1 parent dc1c849 commit da82aef

15 files changed

+58
-57
lines changed

pandas/compat/numpy/function.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,7 @@
2121
from distutils.version import LooseVersion
2222
from typing import Any, Dict, Optional, Union
2323

24-
from numpy import __version__ as _np_version, ndarray
24+
from numpy import __version__, ndarray
2525

2626
from pandas._libs.lib import is_bool, is_integer
2727
from pandas.errors import UnsupportedFunctionCall
@@ -122,7 +122,7 @@ def validate_argmax_with_skipna(skipna, args, kwargs):
122122
ARGSORT_DEFAULTS["kind"] = "quicksort"
123123
ARGSORT_DEFAULTS["order"] = None
124124

125-
if LooseVersion(_np_version) >= LooseVersion("1.17.0"):
125+
if LooseVersion(__version__) >= LooseVersion("1.17.0"):
126126
# GH-26361. NumPy added radix sort and changed default to None.
127127
ARGSORT_DEFAULTS["kind"] = None
128128

pandas/core/algorithms.py

+3-3
Original file line numberDiff line numberDiff line change
@@ -262,7 +262,7 @@ def _get_values_for_rank(values):
262262
return values
263263

264264

265-
def _get_data_algo(values):
265+
def get_data_algo(values):
266266
values = _get_values_for_rank(values)
267267

268268
ndtype = _check_object_for_strings(values)
@@ -491,7 +491,7 @@ def factorize_array(
491491
codes : ndarray
492492
uniques : ndarray
493493
"""
494-
hash_klass, values = _get_data_algo(values)
494+
hash_klass, values = get_data_algo(values)
495495

496496
table = hash_klass(size_hint or len(values))
497497
uniques, codes = table.factorize(
@@ -2086,7 +2086,7 @@ def sort_mixed(values):
20862086

20872087
if sorter is None:
20882088
# mixed types
2089-
hash_klass, values = _get_data_algo(values)
2089+
hash_klass, values = get_data_algo(values)
20902090
t = hash_klass(len(values))
20912091
t.map_locations(values)
20922092
sorter = ensure_platform_int(t.lookup(ordered))

pandas/core/arrays/categorical.py

+3-3
Original file line numberDiff line numberDiff line change
@@ -43,7 +43,7 @@
4343
from pandas.core import ops
4444
from pandas.core.accessor import PandasDelegate, delegate_names
4545
import pandas.core.algorithms as algorithms
46-
from pandas.core.algorithms import _get_data_algo, factorize, take_1d, unique1d
46+
from pandas.core.algorithms import factorize, get_data_algo, take_1d, unique1d
4747
from pandas.core.arrays._mixins import NDArrayBackedExtensionArray
4848
from pandas.core.base import ExtensionArray, NoNewAttributesMixin, PandasObject
4949
import pandas.core.common as com
@@ -2531,8 +2531,8 @@ def _get_codes_for_values(values, categories):
25312531

25322532
# Only hit here when we've already coerced to object dtypee.
25332533

2534-
hash_klass, vals = _get_data_algo(values)
2535-
_, cats = _get_data_algo(categories)
2534+
hash_klass, vals = get_data_algo(values)
2535+
_, cats = get_data_algo(categories)
25362536
t = hash_klass(len(cats))
25372537
t.map_locations(cats)
25382538
return coerce_indexer_dtype(t.lookup(vals), cats)

pandas/core/ops/__init__.py

+10-10
Original file line numberDiff line numberDiff line change
@@ -324,12 +324,12 @@ def _align_method_SERIES(left: "Series", right, align_asobject: bool = False):
324324
return left, right
325325

326326

327-
def _arith_method_SERIES(cls, op, special):
327+
def arith_method_SERIES(cls, op, special):
328328
"""
329329
Wrapper function for Series arithmetic operations, to avoid
330330
code duplication.
331331
"""
332-
assert special # non-special uses _flex_method_SERIES
332+
assert special # non-special uses flex_method_SERIES
333333
op_name = _get_op_name(op, special)
334334

335335
@unpack_zerodim_and_defer(op_name)
@@ -348,12 +348,12 @@ def wrapper(left, right):
348348
return wrapper
349349

350350

351-
def _comp_method_SERIES(cls, op, special):
351+
def comp_method_SERIES(cls, op, special):
352352
"""
353353
Wrapper function for Series arithmetic operations, to avoid
354354
code duplication.
355355
"""
356-
assert special # non-special uses _flex_method_SERIES
356+
assert special # non-special uses flex_method_SERIES
357357
op_name = _get_op_name(op, special)
358358

359359
@unpack_zerodim_and_defer(op_name)
@@ -375,12 +375,12 @@ def wrapper(self, other):
375375
return wrapper
376376

377377

378-
def _bool_method_SERIES(cls, op, special):
378+
def bool_method_SERIES(cls, op, special):
379379
"""
380380
Wrapper function for Series arithmetic operations, to avoid
381381
code duplication.
382382
"""
383-
assert special # non-special uses _flex_method_SERIES
383+
assert special # non-special uses flex_method_SERIES
384384
op_name = _get_op_name(op, special)
385385

386386
@unpack_zerodim_and_defer(op_name)
@@ -398,7 +398,7 @@ def wrapper(self, other):
398398
return wrapper
399399

400400

401-
def _flex_method_SERIES(cls, op, special):
401+
def flex_method_SERIES(cls, op, special):
402402
assert not special # "special" also means "not flex"
403403
name = _get_op_name(op, special)
404404
doc = _make_flex_doc(name, "series")
@@ -614,7 +614,7 @@ def _maybe_align_series_as_frame(frame: "DataFrame", series: "Series", axis: int
614614
return type(frame)(rvalues, index=frame.index, columns=frame.columns)
615615

616616

617-
def _arith_method_FRAME(cls: Type["DataFrame"], op, special: bool):
617+
def arith_method_FRAME(cls: Type["DataFrame"], op, special: bool):
618618
# This is the only function where `special` can be either True or False
619619
op_name = _get_op_name(op, special)
620620
default_axis = _get_frame_op_default_axis(op_name)
@@ -666,7 +666,7 @@ def f(self, other, axis=default_axis, level=None, fill_value=None):
666666
return f
667667

668668

669-
def _flex_comp_method_FRAME(cls: Type["DataFrame"], op, special: bool):
669+
def flex_comp_method_FRAME(cls: Type["DataFrame"], op, special: bool):
670670
assert not special # "special" also means "not flex"
671671
op_name = _get_op_name(op, special)
672672
default_axis = _get_frame_op_default_axis(op_name)
@@ -690,7 +690,7 @@ def f(self, other, axis=default_axis, level=None):
690690
return f
691691

692692

693-
def _comp_method_FRAME(cls: Type["DataFrame"], op, special: bool):
693+
def comp_method_FRAME(cls: Type["DataFrame"], op, special: bool):
694694
assert special # "special" also means "not flex"
695695
op_name = _get_op_name(op, special)
696696

pandas/core/ops/methods.py

+17-17
Original file line numberDiff line numberDiff line change
@@ -44,28 +44,28 @@ def _get_method_wrappers(cls):
4444
# TODO: make these non-runtime imports once the relevant functions
4545
# are no longer in __init__
4646
from pandas.core.ops import (
47-
_arith_method_FRAME,
48-
_arith_method_SERIES,
49-
_bool_method_SERIES,
50-
_comp_method_FRAME,
51-
_comp_method_SERIES,
52-
_flex_comp_method_FRAME,
53-
_flex_method_SERIES,
47+
arith_method_FRAME,
48+
arith_method_SERIES,
49+
bool_method_SERIES,
50+
comp_method_FRAME,
51+
comp_method_SERIES,
52+
flex_comp_method_FRAME,
53+
flex_method_SERIES,
5454
)
5555

5656
if issubclass(cls, ABCSeries):
5757
# Just Series
58-
arith_flex = _flex_method_SERIES
59-
comp_flex = _flex_method_SERIES
60-
arith_special = _arith_method_SERIES
61-
comp_special = _comp_method_SERIES
62-
bool_special = _bool_method_SERIES
58+
arith_flex = flex_method_SERIES
59+
comp_flex = flex_method_SERIES
60+
arith_special = arith_method_SERIES
61+
comp_special = comp_method_SERIES
62+
bool_special = bool_method_SERIES
6363
elif issubclass(cls, ABCDataFrame):
64-
arith_flex = _arith_method_FRAME
65-
comp_flex = _flex_comp_method_FRAME
66-
arith_special = _arith_method_FRAME
67-
comp_special = _comp_method_FRAME
68-
bool_special = _arith_method_FRAME
64+
arith_flex = arith_method_FRAME
65+
comp_flex = flex_comp_method_FRAME
66+
arith_special = arith_method_FRAME
67+
comp_special = comp_method_FRAME
68+
bool_special = arith_method_FRAME
6969
return arith_flex, comp_flex, arith_special, comp_special, bool_special
7070

7171

pandas/io/excel/__init__.py

+4-4
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,9 @@
11
from pandas.io.excel._base import ExcelFile, ExcelWriter, read_excel
2-
from pandas.io.excel._odswriter import _ODSWriter
3-
from pandas.io.excel._openpyxl import _OpenpyxlWriter
2+
from pandas.io.excel._odswriter import ODSWriter as _ODSWriter
3+
from pandas.io.excel._openpyxl import OpenpyxlWriter as _OpenpyxlWriter
44
from pandas.io.excel._util import register_writer
5-
from pandas.io.excel._xlsxwriter import _XlsxWriter
6-
from pandas.io.excel._xlwt import _XlwtWriter
5+
from pandas.io.excel._xlsxwriter import XlsxWriter as _XlsxWriter
6+
from pandas.io.excel._xlwt import XlwtWriter as _XlwtWriter
77

88
__all__ = ["read_excel", "ExcelWriter", "ExcelFile"]
99

pandas/io/excel/_base.py

+8-8
Original file line numberDiff line numberDiff line change
@@ -844,16 +844,16 @@ class ExcelFile:
844844
- ``pyxlsb`` supports Binary Excel files.
845845
"""
846846

847-
from pandas.io.excel._odfreader import _ODFReader
848-
from pandas.io.excel._openpyxl import _OpenpyxlReader
849-
from pandas.io.excel._pyxlsb import _PyxlsbReader
850-
from pandas.io.excel._xlrd import _XlrdReader
847+
from pandas.io.excel._odfreader import ODFReader
848+
from pandas.io.excel._openpyxl import OpenpyxlReader
849+
from pandas.io.excel._pyxlsb import PyxlsbReader
850+
from pandas.io.excel._xlrd import XlrdReader
851851

852852
_engines: Mapping[str, Any] = {
853-
"xlrd": _XlrdReader,
854-
"openpyxl": _OpenpyxlReader,
855-
"odf": _ODFReader,
856-
"pyxlsb": _PyxlsbReader,
853+
"xlrd": XlrdReader,
854+
"openpyxl": OpenpyxlReader,
855+
"odf": ODFReader,
856+
"pyxlsb": PyxlsbReader,
857857
}
858858

859859
def __init__(

pandas/io/excel/_odfreader.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@
1010
from pandas.io.excel._base import BaseExcelReader
1111

1212

13-
class _ODFReader(BaseExcelReader):
13+
class ODFReader(BaseExcelReader):
1414
"""
1515
Read tables out of OpenDocument formatted files.
1616

pandas/io/excel/_odswriter.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@
99
from pandas.io.formats.excel import ExcelCell
1010

1111

12-
class _ODSWriter(ExcelWriter):
12+
class ODSWriter(ExcelWriter):
1313
engine = "odf"
1414
supported_extensions = (".ods",)
1515

pandas/io/excel/_openpyxl.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@
1212
from openpyxl.descriptors.serialisable import Serialisable
1313

1414

15-
class _OpenpyxlWriter(ExcelWriter):
15+
class OpenpyxlWriter(ExcelWriter):
1616
engine = "openpyxl"
1717
supported_extensions = (".xlsx", ".xlsm")
1818

@@ -438,7 +438,7 @@ def write_cells(
438438
setattr(xcell, k, v)
439439

440440

441-
class _OpenpyxlReader(BaseExcelReader):
441+
class OpenpyxlReader(BaseExcelReader):
442442
def __init__(
443443
self,
444444
filepath_or_buffer: FilePathOrBuffer,

pandas/io/excel/_pyxlsb.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66
from pandas.io.excel._base import BaseExcelReader
77

88

9-
class _PyxlsbReader(BaseExcelReader):
9+
class PyxlsbReader(BaseExcelReader):
1010
def __init__(
1111
self,
1212
filepath_or_buffer: FilePathOrBuffer,

pandas/io/excel/_xlrd.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@
88
from pandas.io.excel._base import BaseExcelReader
99

1010

11-
class _XlrdReader(BaseExcelReader):
11+
class XlrdReader(BaseExcelReader):
1212
def __init__(self, filepath_or_buffer, storage_options: StorageOptions = None):
1313
"""
1414
Reader using xlrd engine.

pandas/io/excel/_xlsxwriter.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -158,7 +158,7 @@ def convert(cls, style_dict, num_format_str=None):
158158
return props
159159

160160

161-
class _XlsxWriter(ExcelWriter):
161+
class XlsxWriter(ExcelWriter):
162162
engine = "xlsxwriter"
163163
supported_extensions = (".xlsx",)
164164

pandas/io/excel/_xlwt.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@
99
from xlwt import XFStyle
1010

1111

12-
class _XlwtWriter(ExcelWriter):
12+
class XlwtWriter(ExcelWriter):
1313
engine = "xlwt"
1414
supported_extensions = (".xls",)
1515

pandas/util/_test_decorators.py

+3-2
Original file line numberDiff line numberDiff line change
@@ -33,7 +33,6 @@ def test_foo():
3333

3434
from pandas.compat import IS64, is_platform_windows
3535
from pandas.compat._optional import import_optional_dependency
36-
from pandas.compat.numpy import _np_version
3736

3837
from pandas.core.computation.expressions import NUMEXPR_INSTALLED, USE_NUMEXPR
3938

@@ -205,7 +204,9 @@ def skip_if_no(package: str, min_version: Optional[str] = None):
205204
def skip_if_np_lt(ver_str: str, *args, reason: Optional[str] = None):
206205
if reason is None:
207206
reason = f"NumPy {ver_str} or greater required"
208-
return pytest.mark.skipif(_np_version < LooseVersion(ver_str), *args, reason=reason)
207+
return pytest.mark.skipif(
208+
np.__version__ < LooseVersion(ver_str), *args, reason=reason
209+
)
209210

210211

211212
def parametrize_fixture_doc(*args):

0 commit comments

Comments
 (0)