From 14e819b83b99152b473902f34ba788f98fded4ec Mon Sep 17 00:00:00 2001 From: Gaetan Lepage Date: Tue, 21 Apr 2026 07:48:53 +0000 Subject: [PATCH] python3Packages.skrl: fix jax 0.10.0 compatibility --- .../development/python-modules/skrl/default.nix | 17 +++++++++++++++++ 1 file changed, 17 insertions(+) diff --git a/pkgs/development/python-modules/skrl/default.nix b/pkgs/development/python-modules/skrl/default.nix index 0b37d4d0261b..2990385ef1ab 100644 --- a/pkgs/development/python-modules/skrl/default.nix +++ b/pkgs/development/python-modules/skrl/default.nix @@ -37,6 +37,23 @@ buildPythonPackage (finalAttrs: { hash = "sha256-5lkoYAmMIWqK3+E3WxXMWS9zal2DhZkfl30EkrHKpdI="; }; + # Fix Jax 0.10.0 compatibility + # TypeError: clip() got an unexpected keyword argument 'a_min' + postPatch = '' + substituteInPlace skrl/models/jax/gaussian.py \ + --replace-fail \ + "jnp.clip(log_std, a_min=log_std_min, a_max=log_std_max)" \ + "jnp.clip(log_std, min=log_std_min, max=log_std_max)" \ + --replace-fail \ + "jnp.clip(actions, a_min=clip_actions_min, a_max=clip_actions_max)" \ + "jnp.clip(actions, min=clip_actions_min, max=clip_actions_max)" + + substituteInPlace skrl/models/jax/deterministic.py \ + --replace-fail \ + "jnp.clip(actions, a_min=self._d_clip_actions_min, a_max=self._d_clip_actions_max)" \ + "jnp.clip(actions, min=self._d_clip_actions_min, max=self._d_clip_actions_max)" + ''; + build-system = [ setuptools ]; dependencies = [