File tree Expand file tree Collapse file tree 1 file changed +9
-14
lines changed
Expand file tree Collapse file tree 1 file changed +9
-14
lines changed Original file line number Diff line number Diff line change 77
88
99class MaskedMixin :
10-
11- def __init__ (self ):
12- """Initialize the MaskedMixin class.
13-
14- Note:
15- This class is designed to be used as a mixin for other classes.
16- It provides functionality to apply masking to data.
17- The class should be initialized with the `__init__` method of the
18- parent class to ensure proper initialization.
19- The `set_masks` method should be called to set the masking types
20- and their corresponding probabilities.
21- """
22- # Initialization so that no maskins is applied
23- self .masks = [] # a list of Mask instances
10+ """A mixin class for applying masking to data.
11+
12+ Note:
13+ This class is designed to be used as a mixin for other classes.
14+ It provides functionality to apply masking to data.
15+ The `set_masks` method should be called to set the masking types
16+ and their corresponding probabilities.
17+ """
18+ masks = [] # a list of Mask instances
2419
2520 def set_masks (self , masking : Optional [Dict [str , float ]] = None ) -> None :
2621 """Set the mask type and probability for the dataset.
You can’t perform that action at this time.
0 commit comments