| COLLATORS = {} | |
| def register_collator(name): | |
| def register_collator_cls(cls): | |
| if name in COLLATORS: | |
| return COLLATORS[name] | |
| COLLATORS[name] = cls | |
| return cls | |
| return register_collator_cls | |
| from .tuwen_tbstars import TuwenTBStarsInfonceDataCollator | |
| COLLATORS = {} | |
| def register_collator(name): | |
| def register_collator_cls(cls): | |
| if name in COLLATORS: | |
| return COLLATORS[name] | |
| COLLATORS[name] = cls | |
| return cls | |
| return register_collator_cls | |
| from .tuwen_tbstars import TuwenTBStarsInfonceDataCollator | |