Skip to content

Commit f3e22bc

Browse files
committed
docstrings pt 7
1 parent be4b8e5 commit f3e22bc

File tree

1 file changed

+19
-25
lines changed

1 file changed

+19
-25
lines changed

molpipeline/error_handling.py

Lines changed: 19 additions & 25 deletions
Original file line numberDiff line numberDiff line change
@@ -27,7 +27,6 @@
2727
_S = TypeVar("_S")
2828

2929

30-
# pylint: disable=R0903
3130
class ErrorFilter(ABCPipelineElement):
3231
"""Collects None values and can fill Dummy values to matrices where None values were removed."""
3332

@@ -48,20 +47,17 @@ def __init__(
4847
4948
Parameters
5049
----------
51-
element_ids: list[str]
50+
element_ids: list[str] | None, optional
5251
List of Pipeline Elements for which InvalidInstances can be removed.
53-
filter_everything: bool, optional (default: True)
52+
filter_everything: bool, default=True
5453
If True, element_ids are ignored and all InvalidInstances are removed.
55-
name: str, optional (default: "ErrorFilter")
54+
name: str, default="ErrorFilter"
5655
Name of the pipeline element.
57-
n_jobs: int, optional (default: 1)
56+
n_jobs: int, default=1
5857
Number of parallel jobs to use.
59-
uuid: str, optional (default: None)
58+
uuid: str | None, optional
6059
UUID of the pipeline element.
6160
62-
Returns
63-
-------
64-
None
6561
"""
6662
super().__init__(name=name, n_jobs=n_jobs, uuid=uuid)
6763
self.error_indices = []
@@ -92,16 +88,16 @@ def from_element_list(
9288
----------
9389
element_list: Iterable[TransformingPipelineElement]
9490
List of Pipeline Elements for which None can be removed.
95-
name: str, optional (default: "ErrorFilter")
91+
name: str, default="ErrorFilter"
9692
Name of the pipeline element.
97-
n_jobs: int, optional (default: 1)
93+
n_jobs: int, default=1
9894
Number of parallel jobs to use.
99-
uuid: str, optional (default: None)
95+
uuid: str | None, optional
10096
UUID of the pipeline element.
10197
10298
Returns
10399
-------
104-
ErrorFilter
100+
Self
105101
Constructed ErrorFilter object.
106102
"""
107103
element_ids = {element.uuid for element in element_list}
@@ -173,9 +169,7 @@ def check_removal(self, value: Any) -> bool:
173169
return True
174170
return False
175171

176-
def fit(
177-
self, values: AnyVarSeq, labels: Any = None
178-
) -> Self: # pylint: disable=unused-argument
172+
def fit(self, values: AnyVarSeq, labels: Any = None) -> Self:
179173
"""Fit to input values.
180174
181175
Only for compatibility with sklearn Pipelines.
@@ -438,11 +432,11 @@ def __init__(
438432
Id of the ErrorFilter to use for filling removed values.
439433
fill_value: Any
440434
Value which is used to fill removed values.
441-
name: str, optional (default: "FilterReinserter")
435+
name: str, default="FilterReinserter"
442436
Name of the pipeline element.
443-
n_jobs: int, optional (default: 1)
437+
n_jobs: int, default=1
444438
Number of parallel jobs to use.
445-
uuid: str, optional (default: None)
439+
uuid: str | None, optional
446440
UUID of the pipeline element.
447441
"""
448442
super().__init__(name=name, n_jobs=n_jobs, uuid=uuid)
@@ -467,16 +461,16 @@ def from_error_filter(
467461
ErrorFilter to use for filling removed values.
468462
fill_value: Any
469463
Value which is used to fill removed values.
470-
name: str, optional (default: "FilterReinserter")
464+
name: str, default="FilterReinserter"
471465
Name of the pipeline element.
472-
n_jobs: int, optional (default: 1)
466+
n_jobs: int, default=1
473467
Number of parallel jobs to use.
474-
uuid: str, optional (default: None)
468+
uuid: str | None, optional
475469
UUID of the pipeline element.
476470
477471
Returns
478472
-------
479-
FilterReinserter
473+
Self
480474
Constructed FilterReinserter object.
481475
"""
482476
filler = cls(
@@ -524,7 +518,7 @@ def set_params(self, **parameters: Any) -> Self:
524518
525519
Returns
526520
-------
527-
self
521+
Self
528522
The instance itself.
529523
"""
530524
parameter_copy = dict(parameters)
@@ -667,7 +661,7 @@ def pretransform_single(self, value: Any) -> Any:
667661
def transform(
668662
self,
669663
values: TypeFixedVarSeq,
670-
**params: Any, # pylint: disable=unused-argument
664+
**params: Any,
671665
) -> TypeFixedVarSeq:
672666
"""Transform iterable of values by removing invalid instances.
673667

0 commit comments

Comments
 (0)