python310Packages.dm-sonnet: add missing input

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