diff --git a/modules/estrategia.py b/modules/estrategia.py index e0f9ce6..225c46f 100644 --- a/modules/estrategia.py +++ b/modules/estrategia.py @@ -15,7 +15,7 @@ class TiposEstrategia(Enum): # lista de estratégias. cada entrada é um dicionário com nome de estratégia # e respectivos tracos de personalidade -lista_estrategias: dict[str, float] = {} +lista_estrategias: dict[str, dict[str, float]] = {} def carregarFicheirosEstrategias(pasta_estrategias: str) -> None: @@ -30,8 +30,8 @@ def listaNomesEstrategias() -> list[str]: return list(lista_estrategias.keys()) -def devolvePersonalidade(nome_estrategia: str | None) -> dict[str, float]: +def devolvePersonalidade(nome_estrategia: str | None) -> dict[str, float] | None: if isinstance(nome_estrategia, NoneType): raise ValueError("estratégia {nome_estrategia=} não é válida") - personalidade: dict[str, float] = lista_estrategias.get(nome_estrategia) + personalidade: dict[str, float] | None = lista_estrategias.get(nome_estrategia) return personalidade