Python - mesa: Как получить переменные экземпляра для каждого объекта

Question

Я использую mesa для своей программы. Я пытаюсь выполнить свой класс модели, но я получил AttributeError из класса агента.

Это мой сценарий:

class ComtrModel (Model): 
    """ A model with some number of Agents""" 
    def __init__(self,N): 
     self.num_agents = N 
     self.schedule = RandomActivation(self) 
     for i in range (N): 
      a = CommuterAgent(i) 
      self.schedule.add(a) 

class CommuterAgent (Agent): 
    def __init__(self, model): 
     self.famsize = famsize 
     self.distance = distance 
     self.update_need = None 


    def step(self): 
     if self.distance >= 10000: 
      self.update_need = self.update_need() 
      return 

    def update_need (self, famsize): 
     if self.famsize : 
      self.famsize = famsize 
      return 
      prob_need() 

Как получить переменные каждый агент? Мне нужно проверить его, чтобы убедиться, что модель работает правильно.

До сих пор это мой код для выполнения (на интерактивной сессии):

from src.ComtrModel import * 
model = ComtrModel(5) 
for i in range (10): 
    model.step() 

for key, value in CommuterAgent.step(model): 
    print(key, value) 

EDIT: Но она возвращает

Traceback (most recent call last): 
    File "C:src/__init__.py", line 3, in <module> 
    from src.ComtrModel import * 
    File "C:\src\__init__.py", line 9, in <module> 
    for key, value in CommuterAgent.step(model): 
    File "C:\src\ComtrModel.py", line 40, in step 
    if self.distance >= 10000: 
AttributeError: 'ComtrModel' object has no attribute 'distance' 

Я также пытался что-то вроде этого:

>>> hi_obj = hi() 
>>> hi_obj.__dict__.keys() 

Но он работает только для одного объекта

Answer 1

def __init__(self, model): 
     self.famsize = famsize 
     self.distance = distance 
     self.update_need = None 

Неправильно выглядит. Вы не пропускаете famsize и расстояние как параметры.