fixed member function name

mlpack · Apr 22, 2024 · 61cecf3 · 61cecf3
1 parent 2b5fde8
commit 61cecf3
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/mlpack/methods/reinforcement_learning/environment/vec_env.hpp b/src/mlpack/methods/reinforcement_learning/environment/vec_env.hpp
@@ -128,9 +128,9 @@ class VecEnv
   {
     std::vector<double> rewards(nEnvs);
     for (size_t i = 0; i < nEnvs; i++)
-      rewards[i] = envs[i].Sample(state.data()[i], 
+      rewards[i] = envs[i].Sample(state.Encode()[i], 
                                    action.action[i],
-                                   nextState.data()[i]);
+                                   nextState.Encode()[i]);
 
     return rewards;
   }
@@ -161,7 +161,7 @@ class VecEnv
   {
     std::vector<bool> isTerminalVec(nEnvs);
     for (size_t i = 0; i < nEnvs; i++)
-      isTerminalVec[i] = envs[i].IsTerminal(state.data()[i]);
+      isTerminalVec[i] = envs[i].IsTerminal(state.Encode()[i]);
 
     return isTerminalVec;
   }