add name attribute to Agent

pomdp_py/framework/basics.pxd

@@ -37,6 +37,7 @@ cdef class Agent:
     cdef GenerativeDistribution _cur_belief
     cdef tuple _history
     cdef dict __dict__
+    cdef str _name
 
 cdef class Environment:
     cdef State _init_state

pomdp_py/framework/basics.pyx

@@ -327,25 +327,28 @@ cdef class Agent:
     OR :class:`BlackboxModel` to the planner or the belief update algorithm.
 
     __init__(self, init_belief,
-             policy_model,
+             policy_model=None,
              transition_model=None,
              observation_model=None,
              reward_model=None,
-             blackbox_model=None)
+             blackbox_model=None,
+             name=None)
     """
     def __init__(self, init_belief,
                  policy_model=None,
                  transition_model=None,
                  observation_model=None,
                  reward_model=None,
-                 blackbox_model=None):
+                 blackbox_model=None,
+                 name=None):
         self._init_belief = init_belief
         self._policy_model = policy_model
 
         self._transition_model = transition_model
         self._observation_model = observation_model
         self._reward_model = reward_model
         self._blackbox_model = blackbox_model
+        self._name = name
 
         # For online planning
         self._cur_belief = init_belief
@@ -415,6 +418,15 @@ cdef class Agent:
     def generative_model(self):
         return self.blackbox_model
 
+    @property
+    def name(self):
+        return self._name
+
+    def set_name(self, str name):
+        """set_name(self, str name)
+        gives this agent a name"""
+        self._name = name
+
     def set_models(self, transition_model=None, observation_model=None,
                    reward_model=None, blackbox_model=None, policy_model=None):
         """Re-assign the models to be the ones given."""