diff --git a/pkgs/development/python-modules/scikit-survival/default.nix b/pkgs/development/python-modules/scikit-survival/default.nix index b08e0bd6d89b..fd6b8b83d7b0 100644 --- a/pkgs/development/python-modules/scikit-survival/default.nix +++ b/pkgs/development/python-modules/scikit-survival/default.nix @@ -25,7 +25,7 @@ pytestCheckHook, }: -buildPythonPackage rec { +buildPythonPackage (finalAttrs: { pname = "scikit-survival"; version = "0.26.0"; pyproject = true; @@ -33,7 +33,7 @@ buildPythonPackage rec { src = fetchFromGitHub { owner = "sebp"; repo = "scikit-survival"; - tag = "v${version}"; + tag = "v${finalAttrs.version}"; hash = "sha256-xtrGFNRHF8bL8Q82gIQLayuCSDFMrBBkQ63F+Nmbdes="; }; @@ -85,6 +85,10 @@ buildPythonPackage rec { "test_survival_svm" "test_tree" ] + ++ lib.optionals (stdenv.hostPlatform.isLinux && stdenv.hostPlatform.isAarch64) [ + # Flaky numerical assertion (AssertionError) + "test_baseline_predict" + ] ++ lib.optionals (stdenv.hostPlatform.isDarwin && stdenv.hostPlatform.isAarch64) [ # floating point mismatch on aarch64 # 27079905.88052468 to far from 27079905.880496684 @@ -94,8 +98,8 @@ buildPythonPackage rec { meta = { description = "Survival analysis built on top of scikit-learn"; homepage = "https://github.com/sebp/scikit-survival"; - changelog = "https://github.com/sebp/scikit-survival/releases/tag/v${version}"; + changelog = "https://github.com/sebp/scikit-survival/releases/tag/${finalAttrs.src.tag}"; license = lib.licenses.gpl3Plus; maintainers = [ ]; }; -} +})