feat: 使算法能被自动注册

This commit is contained in:
2026-05-04 13:55:57 +08:00
parent e2b9fb94f3
commit ca86b2d8e9
11 changed files with 101 additions and 139 deletions

View File

@@ -5,11 +5,20 @@ from heurams.services.logger import get_logger
logger = get_logger(__name__)
_registry: dict[str, type["BaseAlgorithm"]] = {}
class BaseAlgorithm:
algo_name = "BaseAlgorithm"
desc = "算法基类"
def __init_subclass__(cls, **kwargs):
super().__init_subclass__(**kwargs)
_registry[cls.algo_name] = cls
@classmethod
def get_registry(cls) -> dict[str, type["BaseAlgorithm"]]:
return dict(_registry)
class AlgodataDict(TypedDict):
real_rept: int
rept: int