Add maskedmixin to __all__

CeliaBenquet · CeliaBenquet · commit 63d5a7c2924f · 2025-05-28T10:23:34.000+02:00
diff --git a/cebra/data/masking.py b/cebra/data/masking.py
@@ -21,8 +21,8 @@
 import torch
 
 __all__ = [
-    "Mask", "RandomNeuronMask", "RandomTimestepMask", "NeuronBlockMask",
-    "TimeBlockMask"
+    "MaskedMixin", "Mask", "RandomNeuronMask", "RandomTimestepMask",
+    "NeuronBlockMask", "TimeBlockMask"
 ]
 
 

Original file line number	Diff line number	Diff line change
`@@ -21,8 +21,8 @@`
`21`	`21`	`import torch`
`22`	`22`
`23`	`23`	`__all__ = [`
`24`		`- "Mask", "RandomNeuronMask", "RandomTimestepMask", "NeuronBlockMask",`
`25`		`- "TimeBlockMask"`
	`24`	`+ "MaskedMixin", "Mask", "RandomNeuronMask", "RandomTimestepMask",`
	`25`	`+ "NeuronBlockMask", "TimeBlockMask"`
`26`	`26`	`]`
`27`	`27`
`28`	`28`