Commit e702dff0 authored by Fabian Affolter's avatar Fabian Affolter
Browse files

python310Packages.dm-sonnet: add missing input

parent cad704eb
Loading
Loading
Loading
Loading
+21 −16
Original line number Diff line number Diff line
{ lib
, fetchFromGitHub
, absl-py
, buildPythonPackage
, numpy
, tabulate
, six
, dm-tree
, absl-py
, wrapt
, docutils
, etils
, fetchFromGitHub
, numpy
, pythonOlder
, tabulate
, tensorflow
, tensorflow-datasets }:
, 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}";
    sha256 = "sha256-YSMeH5ZTfP1OdLBepsxXAVczBG/ghSjCWjoz/I+TFl8=";
    hash = "sha256-YSMeH5ZTfP1OdLBepsxXAVczBG/ghSjCWjoz/I+TFl8=";
  };

  buildInputs = [
    absl-py
  propagatedBuildInputs = [
    dm-tree
    etils
    numpy
    six
    tabulate
    wrapt
  ];
  ] ++ etils.optional-dependencies.epath;

  propagatedBuildInputs = [
    tabulate
  passthru.optional-dependencies = {
    tensorflow = [
      tensorflow
    ];
  };

  checkInputs = [
    docutils