Skip to content

Commit

Permalink
Update return type in api.common
Browse files Browse the repository at this point in the history
Co-authored-by: Diego Ferigo <[email protected]>
  • Loading branch information
flferretti and diegoferigo committed Sep 9, 2024
1 parent e692707 commit 2ae3689
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion src/jaxsim/api/common.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,6 +3,7 @@
import dataclasses
import enum
import functools
from collections.abc import Iterator

import jax
import jax.numpy as jnp
Expand Down Expand Up @@ -43,7 +44,7 @@ class ModelDataWithVelocityRepresentation(JaxsimDataclass, abc.ABC):
@contextlib.contextmanager
def switch_velocity_representation(
self, velocity_representation: VelRepr
) -> contextlib.AbstractContextManager[Self]:
) -> Iterator[Self]:
"""
Context manager to temporarily switch the velocity representation.
Expand Down

0 comments on commit 2ae3689

Please sign in to comment.