Unverified Commit b1bbd024 authored by OTABI Tomoya's avatar OTABI Tomoya Committed by GitHub
Browse files

Merge pull request #315102 from GaetanLepage/jaxtyping 

python311Packages.jaxtyping: 0.2.28 -> 0.2.31
parents 704fe573 a1e787b9

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 ];

  };

}

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

+8 −11
Original line number Diff line number Diff line
@@ -5,9 +5,7 @@ 
  fetchFromGitHub,

  hatchling,

  pythonRelaxDepsHook,

  numpy,

  typeguard,

  typing-extensions,

  cloudpickle,

  equinox,

  ipython,
@@ -21,7 +19,7 @@ 
let

  self = buildPythonPackage rec {

    pname = "jaxtyping";

    version = "0.2.28";

    version = "0.2.31";

    pyproject = true;



    disabled = pythonOlder "3.9";
@@ -30,18 +28,16 @@ let 
      owner = "google";

      repo = "jaxtyping";

      rev = "refs/tags/v${version}";

      hash = "sha256-xDFrgPecUIfCACg/xkMQ8G1+6hNiUUDg9eCZKNpNfzs=";

      hash = "sha256-kuGFzp8sDLq6J/qq8ap3lD3n1pABHurXcbRUtDQyWwE=";

    };



    nativeBuildInputs = [

    build-system = [

      hatchling

      pythonRelaxDepsHook

    ];



    propagatedBuildInputs = [

      numpy

    dependencies = [

      typeguard

      typing-extensions

    ];



    pythonRelaxDeps = [ "typeguard" ];
@@ -70,11 +66,12 @@ let 


    pythonImportsCheck = [ "jaxtyping" ];



    meta = with lib; {

    meta = {

      description = "Type annotations and runtime checking for JAX arrays and PyTrees";

      homepage = "https://github.com/google/jaxtyping";

      license = licenses.mit;

      maintainers = with maintainers; [ GaetanLepage ];

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

      license = lib.licenses.mit;

      maintainers = with lib.maintainers; [ GaetanLepage ];

    };

  };

in