python310Packages.dm-sonnet: fix build

This commit is contained in:
Weijia Wang 2023-05-23 13:23:28 +03:00
parent fdd73b14e8
commit a5f795f79e

View file

@ -1,10 +1,11 @@
{ lib { lib
, absl-py
, buildPythonPackage , buildPythonPackage
, click
, dm-tree , dm-tree
, docutils , docutils
, etils , etils
, fetchFromGitHub , fetchFromGitHub
, fetchpatch
, numpy , numpy
, pythonOlder , pythonOlder
, tabulate , tabulate
@ -27,6 +28,14 @@ buildPythonPackage rec {
hash = "sha256-YSMeH5ZTfP1OdLBepsxXAVczBG/ghSjCWjoz/I+TFl8="; hash = "sha256-YSMeH5ZTfP1OdLBepsxXAVczBG/ghSjCWjoz/I+TFl8=";
}; };
patches = [
(fetchpatch {
name = "replace-np-bool-with-np-bool_.patch";
url = "https://github.com/deepmind/sonnet/commit/df5d099d4557a9a81a0eb969e5a81ed917bcd612.patch";
hash = "sha256-s7abl83osD4wa0ZhqgDyjqQ3gagwGYCdQifwFqhNp34=";
})
];
propagatedBuildInputs = [ propagatedBuildInputs = [
dm-tree dm-tree
etils etils
@ -42,7 +51,9 @@ buildPythonPackage rec {
}; };
nativeCheckInputs = [ nativeCheckInputs = [
click
docutils docutils
tensorflow
tensorflow-datasets tensorflow-datasets
]; ];