From 3e7cbd1e74312dbabc7378fdad548f402974a8d6 Mon Sep 17 00:00:00 2001 From: simonsays1980 Date: Wed, 25 Sep 2024 21:50:57 +0200 Subject: [PATCH] [RLlib] Fix action masking example. (#47817) Signed-off-by: ujjawal-khare --- rllib/examples/rl_modules/classes/action_masking_rlm.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/rllib/examples/rl_modules/classes/action_masking_rlm.py b/rllib/examples/rl_modules/classes/action_masking_rlm.py index 992802ebb13a0..439d938946393 100644 --- a/rllib/examples/rl_modules/classes/action_masking_rlm.py +++ b/rllib/examples/rl_modules/classes/action_masking_rlm.py @@ -106,7 +106,7 @@ def compute_values(self, batch: Dict[str, TensorType], embeddings=None): # to the batch to access them in `_forward_train`. batch["action_mask"] = action_mask # Call the super's method to compute values for GAE. - return super().compute_values(batch, embeddings) + return super().compute_values(batch) def _preprocess_batch( self, batch: Dict[str, TensorType], **kwargs