| """Export reusable Cross-Shuffle, MIL pooling and composite bag-processing components.""" | |
| from .cross_shuffle import CrossTokenShuffle | |
| from .mil_head import MILHead as MILHeadLegacy | |
| from .mil_head_beta import MILHead | |
| from .mil_core import ( | |
| CrossShuffleStage, | |
| CrossShuffleTransMIL, | |
| MILCore, | |
| MILHeadStage, | |
| TransStage, | |
| ) | |
| __all__ = [ | |
| "CrossTokenShuffle", | |
| "MILHead", | |
| "MILHeadLegacy", | |
| "MILCore", | |
| "CrossShuffleStage", | |
| "TransStage", | |
| "MILHeadStage", | |
| "CrossShuffleTransMIL", | |
| ] | |