ensure package_mass arg is passed to package on init, rather than bei…

…ng hardcoded to 50 (#91)
proroklab · Mar 8, 2024 · 7dc49f7 · 7dc49f7
1 parent 2fb6350
commit 7dc49f7
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/vmas/scenarios/reverse_transport.py b/vmas/scenarios/reverse_transport.py
@@ -45,7 +45,7 @@ def make_world(self, batch_dim: int, device: torch.device, **kwargs):
             name=f"package {i}",
             collide=True,
             movable=True,
-            mass=50,
+            mass=self.package_mass,
             shape=Box(
                 length=self.package_length,
                 width=self.package_width,

diff --git a/vmas/scenarios/transport.py b/vmas/scenarios/transport.py
@@ -56,7 +56,7 @@ def make_world(self, batch_dim: int, device: torch.device, **kwargs):
                 name=f"package {i}",
                 collide=True,
                 movable=True,
-                mass=50,
+                mass=self.package_mass,
                 shape=Box(length=self.package_length, width=self.package_width),
                 color=Color.RED,
             )