updating expertAgent for grid2op 1.6.4 and expertOp4Grid 0.1.3

Grid2op · Jan 7, 2022 · a3c59d9 · a3c59d9
1 parent ec6e198
commit a3c59d9
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/l2rpn_baselines/ExpertAgent/ExpertAgent.py b/l2rpn_baselines/ExpertAgent/ExpertAgent.py
@@ -117,7 +117,7 @@ def act(self, observation, reward, done=False):
             subsInCooldown = [i for i in range(observation.n_sub) if (observation.time_before_cooldown_sub[
                                                                           i] >= 1)]  # if you can not run an action currently on a susbtation, so no need to simulate on it
 
-            timestepsOverflowAllowed = observation._obs_env.parameters.NB_TIMESTEP_OVERFLOW_ALLOWED
+            timestepsOverflowAllowed = self.observation_space.obs_env.parameters.NB_TIMESTEP_OVERFLOW_ALLOWED
             isManyOverloads = (n_overloads > timestepsOverflowAllowed)
 
             ltcAlreadyConsidered = []
@@ -234,7 +234,7 @@ def save(self, path):
 
     # we order overloads by usage rate but also by criticity giving remaining timesteps for overload before disconnect
     def getRankedOverloads(self, observation):
-        timestepsOverflowAllowed = observation._obs_env.parameters.NB_TIMESTEP_OVERFLOW_ALLOWED
+        timestepsOverflowAllowed = self.observation_space.obs_env.parameters.NB_TIMESTEP_OVERFLOW_ALLOWED
 
         sort_rho = -np.sort(-observation.rho)  # sort in descending order for positive values
         sort_indices = np.argsort(-observation.rho)