From e702dff003c9ec4dd63aa8db2c827900facb8cd9 Mon Sep 17 00:00:00 2001
From: Fabian Affolter <mail@fabian-affolter.ch>
Date: Tue, 8 Nov 2022 14:42:02 +0100
Subject: [PATCH] python310Packages.dm-sonnet: add missing input

---
 .../python-modules/dm-sonnet/default.nix      | 39 +++++++++++--------
 1 file changed, 22 insertions(+), 17 deletions(-)

diff --git a/pkgs/development/python-modules/dm-sonnet/default.nix b/pkgs/development/python-modules/dm-sonnet/default.nix
index f857eac8503b..e089b91eb0f8 100644
--- a/pkgs/development/python-modules/dm-sonnet/default.nix
+++ b/pkgs/development/python-modules/dm-sonnet/default.nix
@@ -1,40 +1,45 @@
 { lib
-, fetchFromGitHub
-, buildPythonPackage
-, numpy
-, tabulate
-, six
-, dm-tree
 , absl-py
-, wrapt
+, buildPythonPackage
+, dm-tree
 , 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
-    tensorflow
-  ];
+  passthru.optional-dependencies = {
+    tensorflow = [
+      tensorflow
+    ];
+  };
 
   checkInputs = [
     docutils