diff --git a/pkgs/development/python-modules/pymanopt/default.nix b/pkgs/development/python-modules/pymanopt/default.nix new file mode 100644 index 000000000000..5763c79a566d --- /dev/null +++ b/pkgs/development/python-modules/pymanopt/default.nix @@ -0,0 +1,39 @@ +{ lib +, fetchFromGitHub +, buildPythonPackage +, numpy +, scipy +, autograd +, nose2 +, pytestCheckHook +}: + +buildPythonPackage rec { + pname = "pymanopt"; + version = "0.2.5"; + + src = fetchFromGitHub { + owner = pname; + repo = pname; + rev = version; + sha256 = "0zk775v281375sangc5qkwrkb8yc9wx1g8b1917s4s8wszzkp8k6"; + }; + + propagatedBuildInputs = [ numpy scipy ]; + checkInputs = [ nose2 autograd pytestCheckHook ]; + + disabledTestPaths = [ + "tests/test_problem.py" + "tests/test_tensorflow.py" + "tests/test_theano.py" + ]; + + pythonImportsCheck = [ "pymanopt" ]; + + meta = { + description = "Python toolbox for optimization on Riemannian manifolds with support for automatic differentiation"; + homepage = "https://www.pymanopt.org/"; + license = lib.licenses.bsd3; + maintainers = with lib.maintainers; [ yl3dy ]; + }; +} diff --git a/pkgs/top-level/python-packages.nix b/pkgs/top-level/python-packages.nix index e64ac018cbbd..d138ed3039cc 100644 --- a/pkgs/top-level/python-packages.nix +++ b/pkgs/top-level/python-packages.nix @@ -6273,6 +6273,8 @@ in { pymaging_png = callPackage ../development/python-modules/pymaging_png { }; + pymanopt = callPackage ../development/python-modules/pymanopt { }; + pymata-express = callPackage ../development/python-modules/pymata-express { }; pymatgen = callPackage ../development/python-modules/pymatgen { };