change q_ranges

pyomeca · Feb 7, 2024 · df25a45 · df25a45
1 parent 5d87bd9
commit df25a45
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/bioptim/models/biorbd/biorbd_model.py b/bioptim/models/biorbd/biorbd_model.py
@@ -741,11 +741,9 @@ def ranges_from_model(self, variable: str):
                         qddot_range for qddot_range in segment.QDDotRanges()
                     ]
 
-        if variable == "q" or variable == "q_roots" or variable == "q_joints":
+        if variable in ["q", "q_roots", "q_joints"]:
             return q_ranges
-        elif (
-            variable == "qdot" or variable == "qdot_roots" or variable == "qdot_joints"
-        ):
+        elif variable in ["qdot", "qdot_roots", "qdot_joints"]:
             return qdot_ranges
         elif variable == "qddot" or variable == "qddot_joints":
             return qddot_ranges