Merge pull request uzh-rpg#78 from MarioBonse/master

Fix bug on extra info
ykeuter · Mar 8, 2021 · 5161d56 · 5161d56
2 parents 512cac0 + feb866a
commit 5161d56
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/flightlib/src/envs/vec_env.cpp b/flightlib/src/envs/vec_env.cpp
@@ -152,7 +152,7 @@ void VecEnv<EnvBase>::perAgentStep(int agent_id, Ref<MatrixRowMajor<>> act,
   done(agent_id) = envs_[agent_id]->isTerminalState(terminal_reward);
 
   envs_[agent_id]->updateExtraInfo();
-  for (int j = 0; j < extra_info.size(); j++)
+  for (int j = 0; j < extra_info.cols(); j++)
     extra_info(agent_id, j) =
       envs_[agent_id]->extra_info_[extra_info_names_[j]];