[RLlib] Fix action masking example. (ray-project#47817)

Signed-off-by: ujjawal-khare <ujjawal.khare@dream11.com>

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