Skip to content

GitLab

Unverified Commit 6856a45a authored by James Lamb's avatar James Lamb Committed by GitHub
Browse files

[python-package] fix mypy error in Dataset.__init_from_list_np2d() (#5826)

parent 7c1e92dd
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
python-package/lightgbm/basic.py
View file @ 6856a45a
...@@ -2008,7 +2008,7 @@ class Dataset: ...@@ -2008,7 +2008,7 @@ class Dataset:
ptr_data = (ctypes.POINTER(ctypes.c_float) * len(mats))() ptr_data = (ctypes.POINTER(ctypes.c_float) * len(mats))()
holders = [] holders = []
type_ptr_data = None type_ptr_data = -1
for i, mat in enumerate(mats): for i, mat in enumerate(mats):
if len(mat.shape) != 2: if len(mat.shape) != 2:
...@@ -2025,7 +2025,7 @@ class Dataset: ...@@ -2025,7 +2025,7 @@ class Dataset:
mats[i] = np.array(mat.reshape(mat.size), dtype=np.float32) mats[i] = np.array(mat.reshape(mat.size), dtype=np.float32)
chunk_ptr_data, chunk_type_ptr_data, holder = _c_float_array(mats[i]) chunk_ptr_data, chunk_type_ptr_data, holder = _c_float_array(mats[i])
if type_ptr_data is not None and chunk_type_ptr_data != type_ptr_data: if type_ptr_data != -1 and chunk_type_ptr_data != type_ptr_data:
raise ValueError('Input chunks must have same type') raise ValueError('Input chunks must have same type')
ptr_data[i] = chunk_ptr_data ptr_data[i] = chunk_ptr_data
type_ptr_data = chunk_type_ptr_data type_ptr_data = chunk_type_ptr_data
......
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