inisis
2024-12-17 c4412495bded97f40c7cdd5ab37cb5f789bfff41
funasr/register.py
@@ -52,13 +52,15 @@
        def decorator(target_class):
            if not hasattr(self, register_tables_key):
                setattr(self, register_tables_key, {})
                logging.info(f"New registry table added: {register_tables_key}")
                logging.debug(f"New registry table added: {register_tables_key}")
            registry = getattr(self, register_tables_key)
            registry_key = key if key is not None else target_class.__name__
            if registry_key in registry:
                print(f"Key {registry_key} already exists in {register_tables_key}, re-register")
                logging.debug(
                    f"Key {registry_key} already exists in {register_tables_key}, re-register"
                )
            registry[registry_key] = target_class