Move contact_model attribute in JaxSimModel

ami-iit · Jun 18, 2024 · 882a4a0 · 882a4a0
1 parent dfebf77
commit 882a4a0
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/src/jaxsim/api/model.py b/src/jaxsim/api/model.py
@@ -34,6 +34,10 @@ class JaxSimModel(JaxsimDataclass):
         default=jaxsim.terrain.FlatTerrain(), repr=False
     )
 
+    contact_model: jaxsim.rbda.ContactModel | None = dataclasses.field(
+        default=None, repr=False
+    )
+
     kin_dyn_parameters: js.kin_dyn_parameters.KynDynParameters | None = (
         dataclasses.field(default=None, repr=False)
     )
@@ -50,10 +54,6 @@ class JaxSimModel(JaxsimDataclass):
     def description(self) -> jaxsim.parsers.descriptions.ModelDescription:
         return self._description.get()
 
-    contact_model: jaxsim.rbda.ContactModel | None = dataclasses.field(
-        default=None, repr=False, compare=False, hash=False
-    )
-
     def __eq__(self, other: JaxSimModel) -> bool:
 
         if not isinstance(other, JaxSimModel):