nathan/nixpkgs
1
0
Fork 0
mirror of https://github.com/NixOS/nixpkgs.git synced 2026-10-31 01:02:20 +00:00
Code Issues Packages Projects Releases Wiki Activity

python3Packages.waymax: fix build

Browse Source
This commit is contained in:
Gaetan Lepage
2026-10-22 18:22:04 +00:00
parent d013596664
commit affcb7df7d
1 changed files with 12 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
pkgs/development/python-modules/waymax/default.nix
View File

@@ -38,6 +38,18 @@ buildPythonPackage (finalAttrs: {
hash = "sha256-vS0ZxLPKcjRzu5QaFApOVYNO5LBNQjop8wzmdEVdMC8=";
};
# jax>=0.12 no longer supports 2D vectors in jnp.cross
# ValueError: Dimension must be 3 for cross product
postPatch = ''
substituteInPlace waymax/metrics/roadgraph.py \
--replace-fail \
"jnp.cross(points_to_edge, nearest_vector_xys)" \
"(points_to_edge[..., 0] * nearest_vector_xys[..., 1] - points_to_edge[..., 1] * nearest_vector_xys[..., 0])" \
--replace-fail \
"jnp.cross(points_to_edge, prior_vector_xys)" \
"(points_to_edge[..., 0] * prior_vector_xys[..., 1] - points_to_edge[..., 1] * prior_vector_xys[..., 0])"
'';
build-system = [ setuptools ];
dependencies = [