1 files changed, 20 insertions, 4 deletions
diff --git a/pkgs/development/python-modules/autograd/default.nix b/pkgs/development/python-modules/autograd/default.nix
index f0b82fa67a44..26f636e85bf0 100644
--- a/pkgs/development/python-modules/autograd/default.nix
+++ b/pkgs/development/python-modules/autograd/default.nix
@@ -1,20 +1,36 @@
-{ lib, buildPythonPackage, fetchPypi, numpy, future }:
+{ lib
+, buildPythonPackage
+, fetchPypi
+, numpy
+, future
+, pythonOlder
+}:
 
 buildPythonPackage rec {
   pname = "autograd";
-  version = "1.5";
+  version = "1.6.2";
+  format = "setuptools";
+
+  disabled = pythonOlder "3.7";
 
   src = fetchPypi {
     inherit pname version;
-    hash = "sha256-2AvSJRVNHbE8tOrM96GMNYvnIJJkG2hxf5b88dFqzQs=";
+    hash = "sha256-hzHgigxOOJ2GlaQAcq2kUSZBwRO2ys6PTPvo636a7es=";
   };
 
-  propagatedBuildInputs = [ numpy future ];
+  propagatedBuildInputs = [
+    numpy
+    future
+  ];
 
   # Currently, the PyPI tarball doesn't contain the tests. When that has been
   # fixed, enable testing. See: https://github.com/HIPS/autograd/issues/404
   doCheck = false;
 
+  pythonImportsCheck = [
+    "autograd"
+  ];
+
   meta = with lib; {
     homepage = "https://github.com/HIPS/autograd";
     description = "Compute derivatives of NumPy code efficiently";