python312Packages.dm-haiku: cleanup & disable on python 3.13 (#385066) (319d366f) · Commits · nix / nixpkgs

pkgs/development/python-modules/dm-haiku/default.nix

+17 −5

Original line number	Diff line number	Diff line
		{
		lib,
		buildPythonPackage,
		pythonAtLeast,
		fetchFromGitHub,
		fetchpatch,

		# build-system
		setuptools,

		# dependencies
		absl-py,
		flax,
		jax,
		jaxlib,
		jmp,
		numpy,
		tabulate,

		# optional-dependencies
		jax,
		flax,

		# tests
		pytest-xdist,
		pytestCheckHook,
		bsuite,
		@@ -30,6 +39,9 @@ let
		version = "0.0.13";
		pyproject = true;

		# ImportError: `haiku.experimental.flax` features require `flax` to be installed.
		disabled = pythonAtLeast "3.13";

		src = fetchFromGitHub {
		owner = "deepmind";
		repo = "dm-haiku";
		@@ -117,11 +129,11 @@ let
		dontInstall = true;
		});

		meta = with lib; {
		meta = {
		description = "Haiku is a simple neural network library for JAX developed by some of the authors of Sonnet";
		homepage = "https://github.com/deepmind/dm-haiku";
		license = licenses.asl20;
		maintainers = with maintainers; [ ndl ];
		license = lib.licenses.asl20;
		maintainers = with lib.maintainers; [ ndl ];
		};
		};
		in