"...git@developer.sourcefind.cn:tianlh/lightgbm-dcu.git" did not exist on "a323c7583ccdffb7aff49e7f9ca72042aca18395"
Unverified Commit 649ef60c authored by Nikita Titov's avatar Nikita Titov Committed by GitHub
Browse files

[python] Fix typehints in Sequence API (#5465)

parent ccc1f89f
...@@ -1308,15 +1308,19 @@ class Dataset: ...@@ -1308,15 +1308,19 @@ class Dataset:
assert sample_cnt == actual_sample_cnt.value assert sample_cnt == actual_sample_cnt.value
return indices return indices
def _init_from_ref_dataset(self, total_nrow: int, ref_dataset: 'Dataset') -> 'Dataset': def _init_from_ref_dataset(
self,
total_nrow: int,
ref_dataset: _DatasetHandle
) -> 'Dataset':
"""Create dataset from a reference dataset. """Create dataset from a reference dataset.
Parameters Parameters
---------- ----------
total_nrow : int total_nrow : int
Number of rows expected to add to dataset. Number of rows expected to add to dataset.
ref_dataset : Dataset ref_dataset : object
Reference dataset to extract meta from. Handle of reference dataset to extract metadata from.
Returns Returns
------- -------
...@@ -1658,7 +1662,7 @@ class Dataset: ...@@ -1658,7 +1662,7 @@ class Dataset:
def __init_from_seqs( def __init_from_seqs(
self, self,
seqs: List[Sequence], seqs: List[Sequence],
ref_dataset: Optional["Dataset"] = None ref_dataset: Optional[_DatasetHandle] = None
) -> "Dataset": ) -> "Dataset":
""" """
Initialize data from list of Sequence objects. Initialize data from list of Sequence objects.
......
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