pythonPackages.optuna: init at 0.13.0

pkgs/development/python-modules/optuna/default.nix

@@ -0,0 +1,81 @@
+{ lib
+, buildPythonPackage
+, fetchPypi
+, pytest
+, mock
+, bokeh
+, plotly
+, chainer
+, xgboost
+, mpi4py
+, lightgbm
+, Keras
+, mxnet
+, scikit-optimize
+, tensorflow
+, sqlalchemy
+, numpy
+, scipy
+, six
+, cliff
+, colorlog
+, pandas
+, alembic
+, typing
+, pythonOlder
+, isPy27
+}:
+
+buildPythonPackage rec {
+  pname = "optuna";
+  version = "0.13.0";
+  disabled = isPy27;
+
+  src = fetchPypi {
+    inherit pname version;
+    sha256 = "915b9d7b28f7f7cdf015d8617c689ca90eda7a5bbd59c5fc232c9eccc9a91585";
+  };
+
+  checkInputs = [
+    pytest
+    mock
+    bokeh
+    plotly
+    chainer
+    xgboost
+    mpi4py
+    lightgbm
+    Keras
+    mxnet
+    scikit-optimize
+    tensorflow
+  ];
+
+  propagatedBuildInputs = [
+    sqlalchemy
+    numpy
+    scipy
+    six
+    cliff
+    colorlog
+    pandas
+    alembic
+  ] ++ lib.optionals (pythonOlder "3.5") [ typing ];
+
+  configurePhase = if !(pythonOlder "3.5") then ''
+    substituteInPlace setup.py \
+      --replace "'typing'" ""
+  '' else "";
+
+  checkPhase = ''
+    pytest --ignore tests/test_cli.py \
+           --ignore tests/integration_tests/test_chainermn.py
+  '';
+
+  meta = with lib; {
+    description = "A hyperparameter optimization framework";
+    homepage = https://optuna.org/;
+    license = licenses.mit;
+    maintainers = [ maintainers.costrouc ];
+  };
+}

pkgs/top-level/python-packages.nix

@@ -1654,6 +1654,8 @@ in {
 
   openidc-client = callPackage ../development/python-modules/openidc-client {};
 
+  optuna = callPackage ../development/python-modules/optuna { };
+
   idna = callPackage ../development/python-modules/idna { };
 
   mahotas = callPackage ../development/python-modules/mahotas { };