jaxsim 0.7.1.dev43__tar.gz → 0.7.1.dev49__tar.gz

{jaxsim-0.7.1.dev43 → jaxsim-0.7.1.dev49}/.github/workflows/gpu_benchmark.yml

@@ -1,6 +1,9 @@
 name: GPU Benchmarks
 
 on:
+  push:
+    branches:
+      - main
   pull_request:
     types: [opened, reopened, synchronize]
   workflow_dispatch:
@@ -46,7 +49,7 @@ jobs:
         uses: actions/cache/restore@v4
         with:
           path: ./cache
-          key: ${{ steps.get-main-branch-sha.outputs.sha }}-${{ runner.os }}-benchmark
+          key: ${{ runner.os }}-benchmark
 
       - name: Ensure version file is written
         run: |
@@ -60,7 +63,6 @@ jobs:
 
       - name: Compare benchmark results with main branch
         uses: benchmark-action/github-action-benchmark@v1.20.4
-        if: steps.cache.outputs.cache-hit == 'true'
         with:
           tool: 'pytest'
           output-file-path: output.json
@@ -106,4 +108,4 @@ jobs:
         if: ${{ github.ref_name == 'main' }}
         with:
           path: ./cache
-          key: ${{ steps.get-main-branch-sha.outputs.sha }}-${{ runner.os }}-benchmark
+          key: ${{ runner.os }}-benchmark

{jaxsim-0.7.1.dev43 → jaxsim-0.7.1.dev49}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: jaxsim
-Version: 0.7.1.dev43
+Version: 0.7.1.dev49
 Summary: A differentiable physics engine and multibody dynamics library for control and robot learning.
 Author-email: Diego Ferigo <dgferigo@gmail.com>, Filippo Luca Ferretti <filippoluca.ferretti@outlook.com>
 Maintainer-email: Filippo Luca Ferretti <filippo.ferretti@iit.it>, Alessandro Croci <alessandro.croci@iit.it>
@@ -31,8 +31,8 @@ Requires-Python: >=3.10
 Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: coloredlogs
-Requires-Dist: jax>=0.4.26
-Requires-Dist: jaxlib>=0.4.26
+Requires-Dist: jax>=0.4.34
+Requires-Dist: jaxlib>=0.4.34
 Requires-Dist: jaxlie>=1.3.0
 Requires-Dist: jax_dataclasses>=1.4.0
 Requires-Dist: pptree

{jaxsim-0.7.1.dev43 → jaxsim-0.7.1.dev49}/environment.yml

@@ -7,8 +7,8 @@ dependencies:
   # ===========================
   - python >= 3.12.0
   - coloredlogs
-  - jax >= 0.4.26
-  - jaxlib >= 0.4.26
+  - jax >= 0.4.34
+  - jaxlib >= 0.4.34
   - jaxlie >= 1.3.0
   - jax-dataclasses >= 1.4.0
   - optax >= 0.2.3

{jaxsim-0.7.1.dev43 → jaxsim-0.7.1.dev49}/pyproject.toml

@@ -46,8 +46,8 @@ classifiers = [
 ]
 dependencies = [
     "coloredlogs",
-    "jax >= 0.4.26",
-    "jaxlib >= 0.4.26",
+    "jax >= 0.4.34",
+    "jaxlib >= 0.4.34",
     "jaxlie >= 1.3.0",
     "jax_dataclasses >= 1.4.0",
     "pptree",

{jaxsim-0.7.1.dev43 → jaxsim-0.7.1.dev49}/src/jaxsim/_version.py

@@ -17,5 +17,5 @@ __version__: str
 __version_tuple__: VERSION_TUPLE
 version_tuple: VERSION_TUPLE
 
-__version__ = version = '0.7.1.dev43'
-__version_tuple__ = version_tuple = (0, 7, 1, 'dev43')
+__version__ = version = '0.7.1.dev49'
+__version_tuple__ = version_tuple = (0, 7, 1, 'dev49')

{jaxsim-0.7.1.dev43 → jaxsim-0.7.1.dev49}/src/jaxsim.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: jaxsim
-Version: 0.7.1.dev43
+Version: 0.7.1.dev49
 Summary: A differentiable physics engine and multibody dynamics library for control and robot learning.
 Author-email: Diego Ferigo <dgferigo@gmail.com>, Filippo Luca Ferretti <filippoluca.ferretti@outlook.com>
 Maintainer-email: Filippo Luca Ferretti <filippo.ferretti@iit.it>, Alessandro Croci <alessandro.croci@iit.it>
@@ -31,8 +31,8 @@ Requires-Python: >=3.10
 Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: coloredlogs
-Requires-Dist: jax>=0.4.26
-Requires-Dist: jaxlib>=0.4.26
+Requires-Dist: jax>=0.4.34
+Requires-Dist: jaxlib>=0.4.34
 Requires-Dist: jaxlie>=1.3.0
 Requires-Dist: jax_dataclasses>=1.4.0
 Requires-Dist: pptree

{jaxsim-0.7.1.dev43 → jaxsim-0.7.1.dev49}/src/jaxsim.egg-info/requires.txt

@@ -1,6 +1,6 @@
 coloredlogs
-jax>=0.4.26
-jaxlib>=0.4.26
+jax>=0.4.34
+jaxlib>=0.4.34
 jaxlie>=1.3.0
 jax_dataclasses>=1.4.0
 pptree

{jaxsim-0.7.1.dev43 → jaxsim-0.7.1.dev49}/tests/test_api_frame.py

@@ -1,7 +1,7 @@
 import jax
 import jax.numpy as jnp
-import jaxlib.xla_extension
 import pytest
+from jax.errors import JaxRuntimeError
 
 import jaxsim.api as js
 from jaxsim import VelRepr
@@ -50,22 +50,22 @@ def test_frame_index(jaxsim_models_types: js.model.JaxSimModel):
     with pytest.raises(ValueError):
         _ = js.frame.name_to_idx(model=model, frame_name="non_existent_frame")
 
-    with pytest.raises(jaxlib.xla_extension.XlaRuntimeError):
+    with pytest.raises(JaxRuntimeError):
         _ = js.frame.idx_to_name(model=model, frame_index=-1)
 
-    with pytest.raises(jaxlib.xla_extension.XlaRuntimeError):
+    with pytest.raises(JaxRuntimeError):
         _ = js.frame.idx_to_name(model=model, frame_index=n_l - 1)
 
-    with pytest.raises(jaxlib.xla_extension.XlaRuntimeError):
+    with pytest.raises(JaxRuntimeError):
         _ = js.frame.idx_to_name(model=model, frame_index=n_l + n_f)
 
-    with pytest.raises(jaxlib.xla_extension.XlaRuntimeError):
+    with pytest.raises(JaxRuntimeError):
         _ = js.frame.idx_of_parent_link(model=model, frame_index=-1)
 
-    with pytest.raises(jaxlib.xla_extension.XlaRuntimeError):
+    with pytest.raises(JaxRuntimeError):
         _ = js.frame.idx_of_parent_link(model=model, frame_index=n_l - 1)
 
-    with pytest.raises(jaxlib.xla_extension.XlaRuntimeError):
+    with pytest.raises(JaxRuntimeError):
         _ = js.frame.idx_of_parent_link(model=model, frame_index=n_l + n_f)
 
 

{jaxsim-0.7.1.dev43 → jaxsim-0.7.1.dev49}/tests/test_api_joint.py

@@ -1,6 +1,6 @@
 import jax.numpy as jnp
-import jaxlib.xla_extension
 import pytest
+from jax.errors import JaxRuntimeError
 
 import jaxsim.api as js
 
@@ -39,7 +39,7 @@ def test_joint_index(
     with pytest.raises(ValueError):
         _ = js.joint.name_to_idx(model=model, joint_name="non_existent_joint")
 
-    with pytest.raises(jaxlib.xla_extension.XlaRuntimeError):
+    with pytest.raises(JaxRuntimeError):
         _ = js.joint.idx_to_name(model=model, joint_index=-1)
 
     with pytest.raises(IndexError):

{jaxsim-0.7.1.dev43 → jaxsim-0.7.1.dev49}/tests/test_api_link.py

@@ -1,7 +1,7 @@
 import jax
 import jax.numpy as jnp
-import jaxlib.xla_extension
 import pytest
+from jax.errors import JaxRuntimeError
 
 import jaxsim.api as js
 import jaxsim.math
@@ -44,7 +44,7 @@ def test_link_index(
     with pytest.raises(ValueError):
         _ = js.link.name_to_idx(model=model, link_name="non_existent_link")
 
-    with pytest.raises(jaxlib.xla_extension.XlaRuntimeError):
+    with pytest.raises(JaxRuntimeError):
         _ = js.link.idx_to_name(model=model, link_index=-1)
 
     with pytest.raises(IndexError):

{jaxsim-0.7.1.dev43 → jaxsim-0.7.1.dev49}/tests/test_exceptions.py

@@ -3,8 +3,8 @@ from contextlib import redirect_stdout
 
 import jax
 import jax.numpy as jnp
-import jaxlib.xla_extension
 import pytest
+from jax.errors import JaxRuntimeError
 
 from jaxsim import exceptions
 
@@ -68,7 +68,7 @@ def test_exceptions_in_jit_functions():
     # Let's trigger a ValueError exception by passing 42.
     data = 42
     with pytest.raises(
-        jaxlib.xla_extension.XlaRuntimeError,
+        JaxRuntimeError,
         match=f"ValueError: Raising ValueError since data={data}",
     ):
         _ = jit_compiled_function(data=data)
@@ -78,7 +78,7 @@ def test_exceptions_in_jit_functions():
     # Let's trigger a RuntimeError exception by passing -42.
     data = -42
     with pytest.raises(
-        jaxlib.xla_extension.XlaRuntimeError,
+        JaxRuntimeError,
         match=f"RuntimeError: Raising RuntimeError since data={data}",
     ):
         _ = jit_compiled_function(data=data)