python312Packages.jax[lib]: 0.5.0 -> 0.5.1 (#384879) (2b0c931f) · Commits · nix / nixpkgs

pkgs/development/python-modules/distrax/default.nix

+3 −0

Original line number	Diff line number	Diff line
		@@ -42,6 +42,9 @@ buildPythonPackage rec {
		# Flaky: AssertionError: 1 not less than 0.7000000000000001
		"test_von_mises_sample_uniform_ks_test"

		# Flaky: AssertionError: Not equal to tolerance
		"test_composite_methods_are_consistent__with_jit"

		# NotImplementedError: Primitive 'square' does not have a registered inverse.
		"test_against_tfp_bijectors_square"
		"test_log_dets_square__with_device"

pkgs/development/python-modules/equinox/default.nix

+8 −0

Original line number	Diff line number	Diff line
		@@ -32,6 +32,14 @@ buildPythonPackage rec {
		hash = "sha256-hor2qw+aTL7yhV53E/y5DUwyDEYJA8RPRS39xxa8xcw=";
		};

		# Relax speed constraints on tests that can fail on busy builders
		postPatch = ''
		substituteInPlace tests/test_while_loop.py \
		--replace-fail "speed < 0.1" "speed < 0.5" \
		--replace-fail "speed < 0.5" "speed < 1" \
		--replace-fail "speed < 1" "speed < 4" \
		'';

		build-system = [ hatchling ];

		dependencies = [

pkgs/development/python-modules/flax/default.nix

+6 −0

Original line number	Diff line number	Diff line
		@@ -84,6 +84,12 @@ buildPythonPackage rec {
		tensorflow
		];

		pytestFlagsArray = [
		# DeprecationWarning: linear_util.wrap_init is missing a DebugInfo object.
		"-W"
		"ignore::DeprecationWarning"
		];

		disabledTestPaths = [
		# Docs test, needs extra deps + we're not interested in it.
		"docs/_ext/codediff_test.py"

pkgs/development/python-modules/jax-cuda12-pjrt/default.nix

+1 −1

Original line number	Diff line number	Diff line
		@@ -39,7 +39,7 @@ let
		srcs = {
		"x86_64-linux" = fetchurl {
		url = "https://storage.googleapis.com/jax-releases/cuda12_plugin/jax_cuda12_pjrt-${version}-py3-none-manylinux2014_x86_64.whl";
		hash = "sha256-0jgzwbiF2WwnZAAOlQUvK1gnx31JLqaPZ+kDoTJlbbs=";
		hash = "sha256-05Xe87NP1oSOEVlu8pdaiV0fUG31EuQbH8XS3lIMjlE=";
		};
		# "aarch64-linux" = fetchurl {
		# url = "https://storage.googleapis.com/jax-releases/cuda12_plugin/jax_cuda12_pjrt-${version}-py3-none-manylinux2014_aarch64.whl";

pkgs/development/python-modules/jax-cuda12-plugin/default.nix

+8 −8

Original line number	Diff line number	Diff line
		@@ -40,42 +40,42 @@ let
		"3.10-x86_64-linux" = getSrcFromPypi {
		platform = "manylinux2014_x86_64";
		dist = "cp310";
		hash = "sha256-D0Q6azcpjt+weW/NvR+GzoWksIS2vT8fUKT7/Wfe2Gs=";
		hash = "sha256-ymCGSgWlzqRK51dthHtHeeTeYYUKmhgjg0H8Q6dY1Vs=";
		};
		"3.10-aarch64-linux" = getSrcFromPypi {
		platform = "manylinux2014_aarch64";
		dist = "cp310";
		hash = "sha256-AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA=";
		hash = "sha256-488emSaNinBBPw+sM1krh2nBPckdch+RxYeEa+nYhAM=";
		};
		"3.11-x86_64-linux" = getSrcFromPypi {
		platform = "manylinux2014_x86_64";
		dist = "cp311";
		hash = "sha256-qYE1oCIwZLj1xoU+It3BpOOGIVLTf7aF8Nve/+DIASI=";
		hash = "sha256-NGVbjq/H2b/sMbB3rBPULgjY7YZV0kFHxa38AVFSaU8=";
		};
		"3.11-aarch64-linux" = getSrcFromPypi {
		platform = "manylinux2014_aarch64";
		dist = "cp311";
		hash = "sha256-AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA=";
		hash = "sha256-ik8Mje8QnXb1uqwbKuRpCURFzS1/vYxZf83WI+sC/1Q=";
		};
		"3.12-x86_64-linux" = getSrcFromPypi {
		platform = "manylinux2014_x86_64";
		dist = "cp312";
		hash = "sha256-QwWN/FZdjJ2mn0fNTkuVxJXxaG8onvRYTCtygD5vFgc=";
		hash = "sha256-Fp22Rbr+whOO2YOvjxTk0RqElyivpXIC55qRBNmJLxY=";
		};
		"3.12-aarch64-linux" = getSrcFromPypi {
		platform = "manylinux2014_aarch64";
		dist = "cp312";
		hash = "sha256-AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA=";
		hash = "sha256-wqhtVj3AR5rLTHmuz/vuwYn6vY6XAP4/cxvwOV4dbBE=";
		};
		"3.13-x86_64-linux" = getSrcFromPypi {
		platform = "manylinux2014_x86_64";
		dist = "cp313";
		hash = "sha256-3zbEsXbi01qCqfOM13zDadJx5gBR43GgqO9FFD+PWLY=";
		hash = "sha256-J0kTjLGHMZBoa7FPMxBIskwXjwXkdIo9L/fSA2c1rT0=";
		};
		"3.13-aarch64-linux" = getSrcFromPypi {
		platform = "manylinux2014_aarch64";
		dist = "cp313";
		hash = "sha256-AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA=";
		hash = "sha256-C7aXl3+NbGOS9WLDQF6D68xvXLfCCBQFCE9tXPoJ4yo=";
		};
		};
		in

Admin message