Update object_base_3d.py

hanruihua · Dec 7, 2024 · 1d1e377 · 1d1e377
1 parent e482f25
commit 1d1e377
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/irsim/world/object_base_3d.py b/irsim/world/object_base_3d.py
@@ -10,8 +10,11 @@
 
 class ObjectBase3D(ObjectBase):
 
-        def __init__(self, **kwargs):
-            super().__init__(**kwargs)
+        vel_shape = (6, 1)
+        state_shape = (6, 1)
+
+        def __init__(self, state=[0, 0, 0, 0, 0, 0], **kwargs):
+            super().__init__(state=state, **kwargs)