python.pkgs.dm-sonnet: init at 1.30
Needed as a dependency of graph_nets.
This commit is contained in:
parent
1d00bb1375
commit
11801a4d47
85
pkgs/development/python-modules/dm-sonnet/default.nix
Normal file
85
pkgs/development/python-modules/dm-sonnet/default.nix
Normal file
@ -0,0 +1,85 @@
|
|||||||
|
{ lib
|
||||||
|
, fetchFromGitHub
|
||||||
|
, buildBazelPackage
|
||||||
|
, buildPythonPackage
|
||||||
|
, git
|
||||||
|
, python
|
||||||
|
, six
|
||||||
|
, absl-py
|
||||||
|
, semantic-version
|
||||||
|
, contextlib2
|
||||||
|
, wrapt
|
||||||
|
, tensorflow
|
||||||
|
, tensorflow-probability
|
||||||
|
}:
|
||||||
|
|
||||||
|
let
|
||||||
|
version = "1.30";
|
||||||
|
|
||||||
|
# first build all binaries and generate setup.py using bazel
|
||||||
|
bazel-build = buildBazelPackage rec {
|
||||||
|
name = "dm-sonnet-bazel-${version}";
|
||||||
|
|
||||||
|
src = fetchFromGitHub {
|
||||||
|
owner = "deepmind";
|
||||||
|
repo = "sonnet";
|
||||||
|
rev = "v${version}";
|
||||||
|
sha256 = "1dli4a4arx2gmb4p676pfibvnpag9f13znisrk9381g7xpqqmaw6";
|
||||||
|
};
|
||||||
|
|
||||||
|
nativeBuildInputs = [
|
||||||
|
git # needed to fetch the bazel deps (protobuf in particular)
|
||||||
|
];
|
||||||
|
|
||||||
|
# see https://github.com/deepmind/sonnet/blob/master/docs/INSTALL.md
|
||||||
|
bazelTarget = ":install";
|
||||||
|
|
||||||
|
fetchAttrs = {
|
||||||
|
sha256 = "1qwq6xp6gdmy8f0k96q3nsgqs56adrbgq39g5smwik6griwfx9mr";
|
||||||
|
};
|
||||||
|
|
||||||
|
buildAttrs = {
|
||||||
|
preBuild = ''
|
||||||
|
patchShebangs .
|
||||||
|
'';
|
||||||
|
|
||||||
|
installPhase = ''
|
||||||
|
# do not generate a wheel, instead just copy the generated files to $out to be installed by buildPythonPackage
|
||||||
|
sed -i 's,.*bdist_wheel.*,cp -rL . "$out"; exit 0,' bazel-bin/install
|
||||||
|
|
||||||
|
# the target directory "dist" does not actually matter since we're not generating a wheel
|
||||||
|
bazel-bin/install dist
|
||||||
|
'';
|
||||||
|
};
|
||||||
|
};
|
||||||
|
|
||||||
|
# now use pip to install the package prepared by bazel
|
||||||
|
in buildPythonPackage rec {
|
||||||
|
pname = "dm-sonnet";
|
||||||
|
inherit version;
|
||||||
|
|
||||||
|
src = bazel-build;
|
||||||
|
|
||||||
|
propagatedBuildInputs = [
|
||||||
|
six
|
||||||
|
absl-py
|
||||||
|
semantic-version
|
||||||
|
contextlib2
|
||||||
|
wrapt
|
||||||
|
tensorflow
|
||||||
|
tensorflow-probability
|
||||||
|
];
|
||||||
|
|
||||||
|
# not sure how to properly run the real test suite -- through bazel?
|
||||||
|
checkPhase = ''
|
||||||
|
${python.interpreter} -c "import sonnet"
|
||||||
|
'';
|
||||||
|
|
||||||
|
meta = with lib; {
|
||||||
|
description = "TensorFlow-based neural network library";
|
||||||
|
homepage = https://sonnet.dev;
|
||||||
|
license = licenses.asl20;
|
||||||
|
maintainers = with maintainers; [ timokau ];
|
||||||
|
platforms = platforms.linux;
|
||||||
|
};
|
||||||
|
}
|
@ -4927,6 +4927,8 @@ in {
|
|||||||
|
|
||||||
snapperGUI = callPackage ../development/python-modules/snappergui { };
|
snapperGUI = callPackage ../development/python-modules/snappergui { };
|
||||||
|
|
||||||
|
dm-sonnet = callPackage ../development/python-modules/dm-sonnet { };
|
||||||
|
|
||||||
uncertainties = callPackage ../development/python-modules/uncertainties { };
|
uncertainties = callPackage ../development/python-modules/uncertainties { };
|
||||||
|
|
||||||
funcy = callPackage ../development/python-modules/funcy { };
|
funcy = callPackage ../development/python-modules/funcy { };
|
||||||
|
Loading…
Reference in New Issue
Block a user