ensure package_mass arg is passed to package on init, rather than being hardcoded to 50 (#91)

kfu02 · web-flow · commit 7dc49f786fa2 · 2024-03-08T12:03:29.000+01:00
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,
             )

Original file line number	Diff line number	Diff line change
`@@ -56,7 +56,7 @@ def make_world(self, batch_dim: int, device: torch.device, **kwargs):`
`56`	`56`	`name=f"package {i}",`
`57`	`57`	`collide=True,`
`58`	`58`	`movable=True,`
`59`		`- mass=50,`
	`59`	`+ mass=self.package_mass,`
`60`	`60`	`shape=Box(length=self.package_length, width=self.package_width),`
`61`	`61`	`color=Color.RED,`
`62`	`62`	`)`