[python-package] add hint on dtypes in Dataset.set_field() (#5787)

microsoft · Mar 16, 2023 · 2fe2bf0 · 2fe2bf0
1 parent 1787892
commit 2fe2bf0
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/python-package/lightgbm/basic.py b/python-package/lightgbm/basic.py
@@ -2356,6 +2356,7 @@ def set_field(
                 ctypes.c_int(0),
                 ctypes.c_int(_FIELD_TYPE_MAPPER[field_name])))
             return self
+        dtype: "np.typing.DTypeLike"
         if field_name == 'init_score':
             dtype = np.float64
             if _is_1d_collection(data):