diff options
Diffstat (limited to 'pkgs/development/python-modules/gradient-utils/default.nix')
-rw-r--r-- | pkgs/development/python-modules/gradient-utils/default.nix | 34 |
1 files changed, 34 insertions, 0 deletions
diff --git a/pkgs/development/python-modules/gradient-utils/default.nix b/pkgs/development/python-modules/gradient-utils/default.nix new file mode 100644 index 00000000000..d7c77dfc893 --- /dev/null +++ b/pkgs/development/python-modules/gradient-utils/default.nix @@ -0,0 +1,34 @@ +{ buildPythonPackage +, fetchPypi +, hyperopt +, lib +, numpy +, prometheus_client +}: + +buildPythonPackage rec { + pname = "gradient-utils"; + version = "0.3.2"; + + src = fetchPypi { + inherit pname version; + sha256 = "0a99yygv30vibfawk6zd1jd6lva8fjnr99l1ahaf0nqjyw6jl4nw"; + }; + + postPatch = '' + sed -i 's/hyperopt==0.1.2/hyperopt>=0.1.2/' setup.py + sed -i 's/numpy==1.18.5/numpy>=1.18.5/' setup.py + ''; + + propagatedBuildInputs = [ hyperopt prometheus_client numpy ]; + + pythonImportsCheck = [ "gradient_utils" ]; + + meta = with lib; { + description = "Gradient ML SDK"; + homepage = "https://github.com/Paperspace/gradient-utils"; + license = licenses.mit; + platforms = platforms.unix; + maintainers = with maintainers; [ freezeboy ]; + }; +} |