27
27
_S = TypeVar ("_S" )
28
28
29
29
30
- # pylint: disable=R0903
31
30
class ErrorFilter (ABCPipelineElement ):
32
31
"""Collects None values and can fill Dummy values to matrices where None values were removed."""
33
32
@@ -48,20 +47,17 @@ def __init__(
48
47
49
48
Parameters
50
49
----------
51
- element_ids: list[str]
50
+ element_ids: list[str] | None, optional
52
51
List of Pipeline Elements for which InvalidInstances can be removed.
53
- filter_everything: bool, optional ( default: True)
52
+ filter_everything: bool, default= True
54
53
If True, element_ids are ignored and all InvalidInstances are removed.
55
- name: str, optional ( default: "ErrorFilter")
54
+ name: str, default= "ErrorFilter"
56
55
Name of the pipeline element.
57
- n_jobs: int, optional ( default: 1)
56
+ n_jobs: int, default=1
58
57
Number of parallel jobs to use.
59
- uuid: str, optional (default: None)
58
+ uuid: str | None, optional
60
59
UUID of the pipeline element.
61
60
62
- Returns
63
- -------
64
- None
65
61
"""
66
62
super ().__init__ (name = name , n_jobs = n_jobs , uuid = uuid )
67
63
self .error_indices = []
@@ -92,16 +88,16 @@ def from_element_list(
92
88
----------
93
89
element_list: Iterable[TransformingPipelineElement]
94
90
List of Pipeline Elements for which None can be removed.
95
- name: str, optional ( default: "ErrorFilter")
91
+ name: str, default= "ErrorFilter"
96
92
Name of the pipeline element.
97
- n_jobs: int, optional ( default: 1)
93
+ n_jobs: int, default=1
98
94
Number of parallel jobs to use.
99
- uuid: str, optional (default: None)
95
+ uuid: str | None, optional
100
96
UUID of the pipeline element.
101
97
102
98
Returns
103
99
-------
104
- ErrorFilter
100
+ Self
105
101
Constructed ErrorFilter object.
106
102
"""
107
103
element_ids = {element .uuid for element in element_list }
@@ -173,9 +169,7 @@ def check_removal(self, value: Any) -> bool:
173
169
return True
174
170
return False
175
171
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 :
179
173
"""Fit to input values.
180
174
181
175
Only for compatibility with sklearn Pipelines.
@@ -438,11 +432,11 @@ def __init__(
438
432
Id of the ErrorFilter to use for filling removed values.
439
433
fill_value: Any
440
434
Value which is used to fill removed values.
441
- name: str, optional ( default: "FilterReinserter")
435
+ name: str, default= "FilterReinserter"
442
436
Name of the pipeline element.
443
- n_jobs: int, optional ( default: 1)
437
+ n_jobs: int, default=1
444
438
Number of parallel jobs to use.
445
- uuid: str, optional (default: None)
439
+ uuid: str | None, optional
446
440
UUID of the pipeline element.
447
441
"""
448
442
super ().__init__ (name = name , n_jobs = n_jobs , uuid = uuid )
@@ -467,16 +461,16 @@ def from_error_filter(
467
461
ErrorFilter to use for filling removed values.
468
462
fill_value: Any
469
463
Value which is used to fill removed values.
470
- name: str, optional ( default: "FilterReinserter")
464
+ name: str, default= "FilterReinserter"
471
465
Name of the pipeline element.
472
- n_jobs: int, optional ( default: 1)
466
+ n_jobs: int, default=1
473
467
Number of parallel jobs to use.
474
- uuid: str, optional (default: None)
468
+ uuid: str | None, optional
475
469
UUID of the pipeline element.
476
470
477
471
Returns
478
472
-------
479
- FilterReinserter
473
+ Self
480
474
Constructed FilterReinserter object.
481
475
"""
482
476
filler = cls (
@@ -524,7 +518,7 @@ def set_params(self, **parameters: Any) -> Self:
524
518
525
519
Returns
526
520
-------
527
- self
521
+ Self
528
522
The instance itself.
529
523
"""
530
524
parameter_copy = dict (parameters )
@@ -667,7 +661,7 @@ def pretransform_single(self, value: Any) -> Any:
667
661
def transform (
668
662
self ,
669
663
values : TypeFixedVarSeq ,
670
- ** params : Any , # pylint: disable=unused-argument
664
+ ** params : Any ,
671
665
) -> TypeFixedVarSeq :
672
666
"""Transform iterable of values by removing invalid instances.
673
667
0 commit comments