Commit a1e787b9 authored by Gaetan Lepage's avatar Gaetan Lepage
Browse files

python311Packages.equinox: fix equinox tests 

https://github.com/patrick-kidger/equinox/pull/742
parent 948886b4

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

+15 −5
Original line number Diff line number Diff line
@@ -3,6 +3,7 @@ 
  buildPythonPackage,

  pythonOlder,

  fetchFromGitHub,

  fetchpatch,

  hatchling,

  jax,

  jaxlib,
@@ -28,9 +29,18 @@ buildPythonPackage rec { 
    hash = "sha256-3OwHND1YEdg/SppqiB7pCdp6v+lYwTbtX07tmyEMWDo=";

  };



  nativeBuildInputs = [ hatchling ];

  patches = [

    # TODO: remove when next release (0.11.5) is out

    (fetchpatch {

      name = "make-tests-pass-with-jaxtyping-0-2-30";

      url = "https://github.com/patrick-kidger/equinox/commit/cf942646cddffd32519d876c653d09e064bd66b8.patch";

      hash = "sha256-q/vbvLhqT4q+BK+q5sPVY5arzXCmH5LWxt4evAwywtM=";

    })

  ];



  build-system = [ hatchling ];



  propagatedBuildInputs = [

  dependencies = [

    jax

    jaxlib

    jaxtyping
@@ -64,11 +74,11 @@ buildPythonPackage rec { 
    "test_backward_nan"

  ];



  meta = with lib; {

  meta = {

    description = "JAX library based around a simple idea: represent parameterised functions (such as neural networks) as PyTrees";

    changelog = "https://github.com/patrick-kidger/equinox/releases/tag/v${version}";

    homepage = "https://github.com/patrick-kidger/equinox";

    license = licenses.asl20;

    maintainers = with maintainers; [ GaetanLepage ];

    license = lib.licenses.asl20;

    maintainers = with lib.maintainers; [ GaetanLepage ];

  };

}