Fixed bug in stop criterium.

Signed-off-by: simonsays1980 <[email protected]>
ray-project · Sep 27, 2024 · 2b04f6c · 2b04f6c
1 parent 330594e
commit 2b04f6c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/rllib/examples/rl_modules/autoregressive_actions_rl_module.py b/rllib/examples/rl_modules/autoregressive_actions_rl_module.py
@@ -102,7 +102,7 @@
     # exceeds 150 in evaluation.
     stop = {
         f"{NUM_ENV_STEPS_SAMPLED_LIFETIME}": 100000,
-        f"{EVALUATION_RESULTS}/{ENV_RUNNER_RESULTS}/{EPISODE_RETURN_MEAN}": 0.012,
+        f"{EVALUATION_RESULTS}/{ENV_RUNNER_RESULTS}/{EPISODE_RETURN_MEAN}": -0.012,
     }
 
     # Run the example (with Tune).