Hi @dejkam,
You are returning an observation that is incompatible with the observation_space you specified. Each agent’s observation should be 3 values in a dictionary two integers and one real value.
To make it multiagent you bead to provide the observation of each agent seperately with an outer dictionary where each key names one agent in the environment.
obs = {
agent1:{Health_status: 1, LSI_level: 4, Nearby_infected: 0.3},
some_other_agent_name: {Health_status:0, LSI_level:2, Nearby_infected: 0.23},
}