python3Packages.ml-dtypes: fix build on aarch64-darwin

2023-09-29 17:38:44 -07:00 · 2023-09-29 17:38:44 -07:00 · 005e4b6983
parent 874b6543b7
commit 005e4b6983
1 changed files with 9 additions and 0 deletions
--- 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" \