nbv_reconstruction/annotations/external_module.py
2024-08-18 00:37:17 +08:00

7 lines
279 B
Python

EXTERNAL_FREEZE_MODULES = set()
def external_freeze(cls):
if not hasattr(cls, 'load') or not callable(getattr(cls, 'load')):
raise TypeError(f"external module <{cls.__name__}> must implement a 'load' method")
EXTERNAL_FREEZE_MODULES.add(cls)
return cls