Skip to content

Commit ecca817

Browse files
committed
Add parametrize to drone_models namespace
1 parent fbf4cc8 commit ecca817

2 files changed

Lines changed: 5 additions & 0 deletions

File tree

drone_models/__init__.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -19,11 +19,15 @@
1919
os.environ["SCIPY_ARRAY_API"] = "1"
2020
import scipy # noqa: F401, ensure scipy uses array API features
2121

22+
from drone_models.core import parametrize
2223
from drone_models.first_principles import dynamics as _first_principles_dynamics
2324
from drone_models.so_rpy import dynamics as _so_rpy_dynamics
2425
from drone_models.so_rpy_rotor import dynamics as _so_rpy_rotor_dynamics
2526
from drone_models.so_rpy_rotor_drag import dynamics as _so_rpy_rotor_drag_dynamics
2627

28+
__all__ = ["parametrize", "available_models", "model_features"]
29+
30+
2731
available_models: dict[str, Callable] = {
2832
"first_principles": _first_principles_dynamics,
2933
"so_rpy": _so_rpy_dynamics,

drone_models/so_rpy/params.toml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,4 @@
1+
# TODO: Double-check yaw parameters. Currently induces weird behavior in sim
12
[cf2x_L250]
23
mass = 0.03454
34
gravity_vec = [0.0, 0.0, -9.81]

0 commit comments

Comments
 (0)