Skip to content

GitLab

Unverified Commit 90047b96 authored by Omar Salman's avatar Omar Salman Committed by GitHub
Browse files

[python-package] prefix basic.convert_from_sliced_object with _ (#5613)

parent ae0cd5df
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
python-package/lightgbm/basic.py
View file @ 90047b96
...@@ -501,7 +501,7 @@ FEATURE_IMPORTANCE_TYPE_MAPPER = {"split": C_API_FEATURE_IMPORTANCE_SPLIT, ...@@ -501,7 +501,7 @@ FEATURE_IMPORTANCE_TYPE_MAPPER = {"split": C_API_FEATURE_IMPORTANCE_SPLIT,
"gain": C_API_FEATURE_IMPORTANCE_GAIN} "gain": C_API_FEATURE_IMPORTANCE_GAIN}
def convert_from_sliced_object(data): def _convert_from_sliced_object(data):
"""Fix the memory of multi-dimensional sliced object.""" """Fix the memory of multi-dimensional sliced object."""
if isinstance(data, np.ndarray) and isinstance(data.base, np.ndarray): if isinstance(data, np.ndarray) and isinstance(data.base, np.ndarray):
if not data.flags.c_contiguous: if not data.flags.c_contiguous:
...@@ -516,7 +516,7 @@ def c_float_array(data): ...@@ -516,7 +516,7 @@ def c_float_array(data):
if _is_1d_list(data): if _is_1d_list(data):
data = np.array(data, copy=False) data = np.array(data, copy=False)
if _is_numpy_1d_array(data): if _is_numpy_1d_array(data):
data = convert_from_sliced_object(data) data = _convert_from_sliced_object(data)
assert data.flags.c_contiguous assert data.flags.c_contiguous
if data.dtype == np.float32: if data.dtype == np.float32:
ptr_data = data.ctypes.data_as(ctypes.POINTER(ctypes.c_float)) ptr_data = data.ctypes.data_as(ctypes.POINTER(ctypes.c_float))
...@@ -536,7 +536,7 @@ def c_int_array(data): ...@@ -536,7 +536,7 @@ def c_int_array(data):
if _is_1d_list(data): if _is_1d_list(data):
data = np.array(data, copy=False) data = np.array(data, copy=False)
if _is_numpy_1d_array(data): if _is_numpy_1d_array(data):
data = convert_from_sliced_object(data) data = _convert_from_sliced_object(data)
assert data.flags.c_contiguous assert data.flags.c_contiguous
if data.dtype == np.int32: if data.dtype == np.int32:
ptr_data = data.ctypes.data_as(ctypes.POINTER(ctypes.c_int32)) ptr_data = data.ctypes.data_as(ctypes.POINTER(ctypes.c_int32))
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment