Skip to content

Commit 1f13645

Browse files
committed
fix:comfort mypy
1 parent f4eb8c1 commit 1f13645

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

enhancers/enhancer_registry.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -38,12 +38,12 @@ def decorator(cls: type[InformationEnhancer]) -> type[InformationEnhancer]:
3838

3939
# 注册到全局注册表
4040
for modality in modalities:
41-
modality = modality.value.lower() # 统一转换为小写
42-
if modality in ENHANCER_REGISTRY:
43-
logger.error(f"覆盖已存在的信息增强器: {modality} -> {cls.__name__}")
44-
raise ValueError(f"尝试覆盖已存在的信息增强器: {modality} -> {cls.__name__}")
45-
ENHANCER_REGISTRY[modality] = cls
46-
logger.info(f"注册信息增强器: {modality} -> {cls.__name__}")
41+
modality_type = modality.value.lower() # 统一转换为小写
42+
if modality_type in ENHANCER_REGISTRY:
43+
logger.error(f"覆盖已存在的信息增强器: {modality_type} -> {cls.__name__}")
44+
raise ValueError(f"尝试覆盖已存在的信息增强器: {modality_type} -> {cls.__name__}")
45+
ENHANCER_REGISTRY[modality_type] = cls
46+
logger.info(f"注册信息增强器: {modality_type} -> {cls.__name__}")
4747

4848
return cls
4949

0 commit comments

Comments
 (0)