diff options
Diffstat (limited to 'pkgs/development/python-modules/optuna/default.nix')
| -rw-r--r-- | pkgs/development/python-modules/optuna/default.nix | 27 |
1 files changed, 19 insertions, 8 deletions
diff --git a/pkgs/development/python-modules/optuna/default.nix b/pkgs/development/python-modules/optuna/default.nix index bc96c4a2d47..26c72899317 100644 --- a/pkgs/development/python-modules/optuna/default.nix +++ b/pkgs/development/python-modules/optuna/default.nix @@ -1,6 +1,6 @@ { lib , buildPythonPackage -, fetchPypi +, fetchFromGitHub , pytest , mock , bokeh @@ -13,6 +13,7 @@ , mxnet , scikit-optimize , tensorflow +, cma , sqlalchemy , numpy , scipy @@ -21,6 +22,7 @@ , colorlog , pandas , alembic +, tqdm , typing , pythonOlder , isPy27 @@ -28,12 +30,14 @@ buildPythonPackage rec { pname = "optuna"; - version = "0.17.1"; + version = "0.19.0"; disabled = isPy27; - src = fetchPypi { - inherit pname version; - sha256 = "3d1d3547340c47f34f3a416a2e0761a0ff887ae8ce06474e84ebcc8600afd438"; + src = fetchFromGitHub { + owner = "optuna"; + repo = pname; + rev = "v${version}"; + sha256 = "179x2lsckpmkrkkdnvvbzky86g1ba882z677qwbayhsc835wbp0y"; }; checkInputs = [ @@ -49,6 +53,7 @@ buildPythonPackage rec { mxnet scikit-optimize tensorflow + cma ]; propagatedBuildInputs = [ @@ -60,16 +65,22 @@ buildPythonPackage rec { colorlog pandas alembic - ] ++ lib.optionals (pythonOlder "3.5") [ typing ]; + tqdm + ] ++ lib.optionals (pythonOlder "3.5") [ + typing + ]; configurePhase = if !(pythonOlder "3.5") then '' substituteInPlace setup.py \ - --replace "'typing'" "" + --replace "'typing'," "" '' else ""; checkPhase = '' pytest --ignore tests/test_cli.py \ - --ignore tests/integration_tests/test_chainermn.py + --ignore tests/integration_tests/test_chainermn.py \ + --ignore tests/integration_tests/test_pytorch_lightning.py \ + --ignore tests/integration_tests/test_pytorch_ignite.py \ + --ignore tests/integration_tests/test_fastai.py ''; meta = with lib; { |