Skip to content

Commit e6c0343

Browse files
committed
Change version to 0.1.0
1 parent 420e77d commit e6c0343

File tree

3 files changed

+9
-5
lines changed

3 files changed

+9
-5
lines changed

pyproject.toml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@ name = "jsrm" # Required
1717
#
1818
# For a discussion on single-sourcing the version, see
1919
# https://packaging.python.org/guides/single-sourcing-package-version/
20-
version = "0.0.17" # Required
20+
version = "0.1.0" # Required
2121

2222
# This is a one-line description or tagline of what your project does. This
2323
# corresponds to the "Summary" metadata field:

src/jsrm/systems/pcs.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -79,6 +79,8 @@ class PCS(eqx.Module):
7979
Computes the dynamical matrix for the system.
8080
resolve_upon_time(q0: Array, qd0: Array, actuation_args: Tuple[Array], t0: float, t1: float, dt: float, skip_steps: int, solver: AbstractSolver, max_steps: Optional[int] = None) -> Tuple[Array, Array, Array]:
8181
Simulates the robot dynamics over time using the specified solver.
82+
forward_dynamics(t: float, y: Array, actuation_args: Optional[Tuple]) -> Array:
83+
Computes the forward dynamics of the system at a given time t.
8284
8385
kinetic_energy(q: Array, qd: Array) -> float:
8486
Computes the kinetic energy of the system.
@@ -1382,7 +1384,7 @@ def operational_space_dynamical_matrices(
13821384
return Lambda, mu, J, J_d, JB_pinv
13831385

13841386
@eqx.filter_jit
1385-
def _forward_dynamics(
1387+
def forward_dynamics(
13861388
self,
13871389
t: float,
13881390
y: Array,
@@ -1458,7 +1460,7 @@ def resolve_upon_time(
14581460
"""
14591461
y0 = jnp.concatenate([q0, qd0]) # Initial state vector
14601462

1461-
term = ODETerm(self._forward_dynamics)
1463+
term = ODETerm(self.forward_dynamics)
14621464

14631465
t = jnp.arange(t0, t1, dt) # Time points for the solution
14641466
saveat = SaveAt(ts=t[::skip_steps]) # Save at specified time points

src/jsrm/systems/planar_pcs.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -79,6 +79,8 @@ class PlanarPCS(eqx.Module):
7979
Computes the dynamical matrix for the system.
8080
resolve_upon_time(q0: Array, qd0: Array, actuation_args: Tuple[Array], t0: float, t1: float, dt: float, skip_steps: int, solver: AbstractSolver, max_steps: Optional[int] = None) -> Tuple[Array, Array, Array]:
8181
Simulates the robot dynamics over time using the specified solver.
82+
forward_dynamics(t: float, y: Array, actuation_args: Optional[Tuple]) -> Array:
83+
Computes the forward dynamics of the system at a given time t.
8284
8385
kinetic_energy(q: Array, qd: Array) -> float:
8486
Computes the kinetic energy of the system.
@@ -1373,7 +1375,7 @@ def operational_space_dynamical_matrices(
13731375
return Lambda, mu, J, J_d, JB_pinv
13741376

13751377
@eqx.filter_jit
1376-
def _forward_dynamics(
1378+
def forward_dynamics(
13771379
self,
13781380
t: float,
13791381
y: Array,
@@ -1449,7 +1451,7 @@ def resolve_upon_time(
14491451
"""
14501452
y0 = jnp.concatenate([q0, qd0]) # Initial state vector
14511453

1452-
term = ODETerm(self._forward_dynamics)
1454+
term = ODETerm(self.forward_dynamics)
14531455

14541456
t = jnp.arange(t0, t1, dt) # Time points for the solution
14551457
saveat = SaveAt(ts=t[::skip_steps]) # Save at specified time points

0 commit comments

Comments
 (0)