final fix in... needed to access invariant of agent id if parent == -1.

mbmcgarry · Feb 12, 2014 · 1e80be3 · 1e80be3
1 parent 7be6b31
commit 1e80be3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/visitors.py b/tests/visitors.py
@@ -98,7 +98,7 @@ def walk(self):
     def visit_agents(self, table):
         d = {self.agent_invariants[row[_agent_key]]:
                  tuple(row[i] if i not in _agent_id_names
-                       else self.agent_invariants[row[i]] 
+                       else self.agent_invariants[row[_agent_key]] 
                        for i in _agent_schema)
              for row in table.iterrows()}
         return tuple((k, d[k]) for k in sorted(d.keys()))