diff --git a/movement_primitives/dmp/_cartesian_dmp.py b/movement_primitives/dmp/_cartesian_dmp.py index 34b167e..29dbce4 100644 --- a/movement_primitives/dmp/_cartesian_dmp.py +++ b/movement_primitives/dmp/_cartesian_dmp.py @@ -265,7 +265,7 @@ def open_loop(self, run_t=None, coupling_term=None, step_function : str, optional (default: 'rk4') DMP integration function. Possible options: 'rk4', 'euler', - 'euler-cython'. + 'euler-cython', 'rk4-cython'. quaternion_step_function : str, optional (default: 'cython' if available) DMP integration function. Possible options: 'python', 'cython'. diff --git a/movement_primitives/dmp/_dmp.py b/movement_primitives/dmp/_dmp.py index 69b2671..2f7ad60 100644 --- a/movement_primitives/dmp/_dmp.py +++ b/movement_primitives/dmp/_dmp.py @@ -291,6 +291,7 @@ def dmp_step_euler( from ..dmp_fast import dmp_step as dmp_step_euler_cython, dmp_step_rk4 as dmp_step_rk4_cython DMP_STEP_FUNCTIONS["euler-cython"] = dmp_step_euler_cython DMP_STEP_FUNCTIONS["rk4-cython"] = dmp_step_rk4_cython + DEFAULT_DMP_STEP_FUNCTION = "rk4-cython" except ImportError: pass @@ -410,7 +411,7 @@ def open_loop(self, run_t=None, coupling_term=None, step_function : str, optional (default: 'rk4') DMP integration function. Possible options: 'rk4', 'euler', - 'euler-cython'. + 'euler-cython', 'rk4-cython'. Returns -------