From 15d7e7cb0768c84e05cc96201f360ad9b0f4d6a3 Mon Sep 17 00:00:00 2001 From: Gaetan Lepage Date: Mon, 5 Jan 2026 11:12:18 +0000 Subject: [PATCH 1/4] python3Packages.jaxtyping: 0.3.4 -> 0.3.5 Diff: https://github.com/google/jaxtyping/compare/v0.3.4...v0.3.5 Changelog: https://github.com/patrick-kidger/jaxtyping/releases/tag/v0.3.5 --- pkgs/development/python-modules/jaxtyping/default.nix | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/pkgs/development/python-modules/jaxtyping/default.nix b/pkgs/development/python-modules/jaxtyping/default.nix index 1193fec64749..d3730fd2b7d6 100644 --- a/pkgs/development/python-modules/jaxtyping/default.nix +++ b/pkgs/development/python-modules/jaxtyping/default.nix @@ -23,14 +23,14 @@ let self = buildPythonPackage rec { pname = "jaxtyping"; - version = "0.3.4"; + version = "0.3.5"; pyproject = true; src = fetchFromGitHub { owner = "google"; repo = "jaxtyping"; tag = "v${version}"; - hash = "sha256-u9JsZm5uz2XXQnhgRe5P/fJrKKnyIMdCTnBLCbW+kAw="; + hash = "sha256-0Vt6UD1xQkwve6yDVi5XQCoJ/IsJWHCkGesj66myQq4="; }; build-system = [ hatchling ]; From 56534f595e437f3f4723f92b2424838670ae1250 Mon Sep 17 00:00:00 2001 From: Gaetan Lepage Date: Mon, 5 Jan 2026 11:15:04 +0000 Subject: [PATCH 2/4] python3Package.jaxtyping: flatten structure --- .../python-modules/jaxtyping/default.nix | 88 +++++++++---------- 1 file changed, 44 insertions(+), 44 deletions(-) diff --git a/pkgs/development/python-modules/jaxtyping/default.nix b/pkgs/development/python-modules/jaxtyping/default.nix index d3730fd2b7d6..68bce01a2803 100644 --- a/pkgs/development/python-modules/jaxtyping/default.nix +++ b/pkgs/development/python-modules/jaxtyping/default.nix @@ -18,58 +18,58 @@ pytestCheckHook, tensorflow, torch, + + # passthru + jaxtyping, }: -let - self = buildPythonPackage rec { - pname = "jaxtyping"; - version = "0.3.5"; - pyproject = true; +buildPythonPackage rec { + pname = "jaxtyping"; + version = "0.3.5"; + pyproject = true; - src = fetchFromGitHub { - owner = "google"; - repo = "jaxtyping"; - tag = "v${version}"; - hash = "sha256-0Vt6UD1xQkwve6yDVi5XQCoJ/IsJWHCkGesj66myQq4="; - }; + src = fetchFromGitHub { + owner = "google"; + repo = "jaxtyping"; + tag = "v${version}"; + hash = "sha256-0Vt6UD1xQkwve6yDVi5XQCoJ/IsJWHCkGesj66myQq4="; + }; - build-system = [ hatchling ]; + build-system = [ hatchling ]; - dependencies = [ - wadler-lindig - ]; + dependencies = [ + wadler-lindig + ]; - pythonImportsCheck = [ "jaxtyping" ]; + pythonImportsCheck = [ "jaxtyping" ]; - nativeCheckInputs = [ - cloudpickle - equinox - ipython - jax - jaxlib - pytestCheckHook - tensorflow - torch - ]; + nativeCheckInputs = [ + cloudpickle + equinox + ipython + jax + jaxlib + pytestCheckHook + tensorflow + torch + ]; - doCheck = false; + doCheck = false; - # Enable tests via passthru to avoid cyclic dependency with equinox. - passthru.tests = { - check = self.overridePythonAttrs { - # We disable tests because they complain about the version of typeguard being too new. - doCheck = false; - catchConflicts = false; - }; - }; - - meta = { - description = "Type annotations and runtime checking for JAX arrays and PyTrees"; - homepage = "https://github.com/google/jaxtyping"; - changelog = "https://github.com/patrick-kidger/jaxtyping/releases/tag/v${version}"; - license = lib.licenses.mit; - maintainers = with lib.maintainers; [ GaetanLepage ]; + # Enable tests via passthru to avoid cyclic dependency with equinox. + passthru.tests = { + check = jaxtyping.overridePythonAttrs { + # We disable tests because they complain about the version of typeguard being too new. + doCheck = false; + catchConflicts = false; }; }; -in -self + + meta = { + description = "Type annotations and runtime checking for JAX arrays and PyTrees"; + homepage = "https://github.com/google/jaxtyping"; + changelog = "https://github.com/patrick-kidger/jaxtyping/releases/tag/v${version}"; + license = lib.licenses.mit; + maintainers = with lib.maintainers; [ GaetanLepage ]; + }; +} From a45d211bec37b06c53a6db92b96b0ebd07bbd87e Mon Sep 17 00:00:00 2001 From: Gaetan Lepage Date: Mon, 5 Jan 2026 11:16:04 +0000 Subject: [PATCH 3/4] python3Package.jaxtyping: switch to finalAttrs --- pkgs/development/python-modules/jaxtyping/default.nix | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/pkgs/development/python-modules/jaxtyping/default.nix b/pkgs/development/python-modules/jaxtyping/default.nix index 68bce01a2803..bb00d1d1a293 100644 --- a/pkgs/development/python-modules/jaxtyping/default.nix +++ b/pkgs/development/python-modules/jaxtyping/default.nix @@ -23,7 +23,7 @@ jaxtyping, }: -buildPythonPackage rec { +buildPythonPackage (finalAttrs: { pname = "jaxtyping"; version = "0.3.5"; pyproject = true; @@ -31,7 +31,7 @@ buildPythonPackage rec { src = fetchFromGitHub { owner = "google"; repo = "jaxtyping"; - tag = "v${version}"; + tag = "v${finalAttrs.version}"; hash = "sha256-0Vt6UD1xQkwve6yDVi5XQCoJ/IsJWHCkGesj66myQq4="; }; @@ -68,8 +68,8 @@ buildPythonPackage rec { meta = { description = "Type annotations and runtime checking for JAX arrays and PyTrees"; homepage = "https://github.com/google/jaxtyping"; - changelog = "https://github.com/patrick-kidger/jaxtyping/releases/tag/v${version}"; + changelog = "https://github.com/patrick-kidger/jaxtyping/releases/tag/${finalAttrs.src.tag}"; license = lib.licenses.mit; maintainers = with lib.maintainers; [ GaetanLepage ]; }; -} +}) From 3be8a68ad3aa8306cad31c702c44072b344a602f Mon Sep 17 00:00:00 2001 From: Gaetan Lepage Date: Mon, 5 Jan 2026 11:18:21 +0000 Subject: [PATCH 4/4] python3Package.jaxtyping: update repo owner --- pkgs/development/python-modules/jaxtyping/default.nix | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/pkgs/development/python-modules/jaxtyping/default.nix b/pkgs/development/python-modules/jaxtyping/default.nix index bb00d1d1a293..8390583865b8 100644 --- a/pkgs/development/python-modules/jaxtyping/default.nix +++ b/pkgs/development/python-modules/jaxtyping/default.nix @@ -29,7 +29,7 @@ buildPythonPackage (finalAttrs: { pyproject = true; src = fetchFromGitHub { - owner = "google"; + owner = "patrick-kidger"; repo = "jaxtyping"; tag = "v${finalAttrs.version}"; hash = "sha256-0Vt6UD1xQkwve6yDVi5XQCoJ/IsJWHCkGesj66myQq4="; @@ -67,7 +67,7 @@ buildPythonPackage (finalAttrs: { meta = { description = "Type annotations and runtime checking for JAX arrays and PyTrees"; - homepage = "https://github.com/google/jaxtyping"; + homepage = "https://github.com/patrick-kidger/jaxtyping"; changelog = "https://github.com/patrick-kidger/jaxtyping/releases/tag/${finalAttrs.src.tag}"; license = lib.licenses.mit; maintainers = with lib.maintainers; [ GaetanLepage ];