amend

facebookresearch · Jul 31, 2024 · c84a9af · c84a9af
1 parent 12fafb8
commit c84a9af
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/benchmarl/algorithms/isac.py b/benchmarl/algorithms/isac.py
@@ -204,8 +204,8 @@ def _get_policy_for_loss(
                 ),
                 distribution_kwargs=(
                     {
-                        "min": self.action_spec[(group, "action")].space.low,
-                        "max": self.action_spec[(group, "action")].space.high,
+                        "low": self.action_spec[(group, "action")].space.low,
+                        "high": self.action_spec[(group, "action")].space.high,
                     }
                     if self.use_tanh_normal
                     else {}

diff --git a/benchmarl/algorithms/maddpg.py b/benchmarl/algorithms/maddpg.py
@@ -125,8 +125,8 @@ def _get_policy_for_loss(
                 distribution_class=TanhDelta if self.use_tanh_mapping else Delta,
                 distribution_kwargs=(
                     {
-                        "min": self.action_spec[(group, "action")].space.low,
-                        "max": self.action_spec[(group, "action")].space.high,
+                        "low": self.action_spec[(group, "action")].space.low,
+                        "high": self.action_spec[(group, "action")].space.high,
                     }
                     if self.use_tanh_mapping
                     else {}