Something went wrong. Try again.
What happens if I push all of nixpkgs
Something went wrong. Try again.
1.3 kB · 70 lines
Nix
at litex
1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071{ lib, buildPythonPackage, click, dm-tree, docutils, etils, fetchFromGitHub, fetchpatch, numpy, pythonOlder, tabulate, tensorflow, tensorflow-datasets, wrapt}:
buildPythonPackage rec { pname = "dm-sonnet"; version = "2.0.0"; format = "setuptools";
disabled = pythonOlder "3.6";
src = fetchFromGitHub { owner = "deepmind"; repo = "sonnet"; rev = "v${version}"; 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 = [ dm-tree etils numpy tabulate wrapt ] ++ etils.optional-dependencies.epath;
passthru.optional-dependencies = { tensorflow = [ tensorflow ]; };
nativeCheckInputs = [ click docutils tensorflow tensorflow-datasets ];
pythonImportsCheck = [ "sonnet" ];
meta = with lib; { description = "Library for building neural networks in TensorFlow"; homepage = "https://github.com/deepmind/sonnet"; license = licenses.asl20; maintainers = with maintainers; [ onny ]; };}