feat(issue-74): Add method for choosing action in Agent class

#74

Author: CinquilCinquil

tests/units/agents/test_agent_base.py

@@ -42,10 +42,12 @@ def test_abstract_methods(self):
 
         # WHEN
         step_return = fake_agent.step()
+        choose_action_return = fake_agent.choose_action(FakeActionSpace())
 
         # THEN
         assert isinstance(AgentBase, ABCMeta)
         assert step_return is None
+        assert choose_action_return is None
 
     def test_reset(self):
         # GIVEN

urnai/agents/agent_base.py

@@ -1,5 +1,6 @@
 from abc import ABC, abstractmethod
 
+from urnai.actions.action_base import ActionBase
 from urnai.actions.action_space_base import ActionSpaceBase
 from urnai.models.model_base import ModelBase
 from urnai.states.state_base import StateBase
@@ -27,6 +28,13 @@ def __init__(self, action_space : ActionSpaceBase,
     def step(self) -> None:
         ...
 
+    @abstractmethod
+    def choose_action(self, action_space : ActionSpaceBase) -> ActionBase:
+        """
+        Method that contains the agent's strategy for choosing actions
+        """
+        ...
+
     def reset(self, episode=0) -> None:
         """
         Resets some Agent class variables, such as previous_action