diff options
Diffstat (limited to 'pkgs/development/python-modules/ml-dtypes/default.nix')
-rw-r--r-- | pkgs/development/python-modules/ml-dtypes/default.nix | 9 |
1 files changed, 9 insertions, 0 deletions
diff --git a/pkgs/development/python-modules/ml-dtypes/default.nix b/pkgs/development/python-modules/ml-dtypes/default.nix index 08c4a02ca90..9b99c06ce40 100644 --- a/pkgs/development/python-modules/ml-dtypes/default.nix +++ b/pkgs/development/python-modules/ml-dtypes/default.nix @@ -2,6 +2,7 @@ , buildPythonPackage , pythonOlder , fetchFromGitHub +, fetchpatch , setuptools , pybind11 , numpy @@ -27,6 +28,14 @@ buildPythonPackage rec { fetchSubmodules = true; }; + patches = [ + # See https://github.com/jax-ml/ml_dtypes/issues/106. + (fetchpatch { + url = "https://github.com/jax-ml/ml_dtypes/commit/c082a2df6bc0686b35c4b4a303fd1990485e181f.patch"; + hash = "sha256-aVJy9vT00b98xOrJCdbCHSZBI3uyjafmN88Z2rjBS48="; + }) + ]; + postPatch = '' substituteInPlace pyproject.toml \ --replace "numpy~=1.21.2" "numpy" \ |