Fix bug on extra info

If the extra info vectore was used it caused a seg fault.
hungpnv · Jan 23, 2021 · feb866a · feb866a
1 parent 8b2983c
commit feb866a
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]];