Skip to content

Commit 57c9494

Browse files
committed
Remove masking init()
1 parent 464f4aa commit 57c9494

File tree

1 file changed

+9
-14
lines changed

1 file changed

+9
-14
lines changed

cebra/data/masking.py

Lines changed: 9 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -7,20 +7,15 @@
77

88

99
class 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.

0 commit comments

Comments
 (0)