Fix types according to mypy

2023-07-14 16:38:55 +01:00
parent e522e56ff1
commit 98ac228f90
8 changed files with 32 additions and 22 deletions
--- a/src/primaite/agents/utils.py
+++ b/src/primaite/agents/utils.py
@@ -34,7 +34,7 @@ def transform_action_node_readable(action: List[int]) -> List[Union[int, str]]:
    else:
        property_action = "NONE"

-    new_action = [action[0], action_node_property, property_action, action[3]]
+    new_action: list[Union[int, str]] = [action[0], action_node_property, property_action, action[3]]
    return new_action