nordic-dev.net/nixpkgs
nordic-dev.net/nixpkgs
2
0
Fork 0
mirror of https://github.com/NixOS/nixpkgs.git synced 2025-02-10 06:04:14 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #315102 from GaetanLepage/jaxtyping

Browse Source 
python311Packages.jaxtyping: 0.2.28 -> 0.2.31
This commit is contained in:
OTABI Tomoya 2024-07-02 19:06:59 +09:00 committed by GitHub
commit b1bbd0246a
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 23 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
pkgs/development/python-modules/equinox/default.nix
View File

@ -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=";
})
];
propagatedBuildInputs = [
build-system = [ hatchling ];
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 ];
};
}

19
pkgs/development/python-modules/jaxtyping/default.nix
View File

@ -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