summaryrefslogtreecommitdiff
path: root/pkgs/development/python-modules/optuna/default.nix
diff options
context:
space:
mode:
Diffstat (limited to 'pkgs/development/python-modules/optuna/default.nix')
-rw-r--r--pkgs/development/python-modules/optuna/default.nix94
1 files changed, 45 insertions, 49 deletions
diff --git a/pkgs/development/python-modules/optuna/default.nix b/pkgs/development/python-modules/optuna/default.nix
index 97f065323c72..86b39519741f 100644
--- a/pkgs/development/python-modules/optuna/default.nix
+++ b/pkgs/development/python-modules/optuna/default.nix
@@ -1,44 +1,45 @@
-{ lib
-, buildPythonPackage
-, fetchFromGitHub
-, pytestCheckHook
-, pythonOlder
-, alembic
-, boto3
-, botorch
-, catboost
-, cma
-, cmaes
-, colorlog
-, distributed
-, fakeredis
-, google-cloud-storage
-, lightgbm
-, matplotlib
-, mlflow
-, moto
-, numpy
-, packaging
-, pandas
-, plotly
-, pytest-xdist
-, pytorch-lightning
-, pyyaml
-, redis
-, scikit-learn
-, scikit-optimize
-, scipy
-, setuptools
-, shap
-, sqlalchemy
-, tensorflow
-, torch
-, torchaudio
-, torchvision
-, tqdm
-, wandb
-, wheel
-, xgboost
+{
+ lib,
+ buildPythonPackage,
+ fetchFromGitHub,
+ pytestCheckHook,
+ pythonOlder,
+ alembic,
+ boto3,
+ botorch,
+ catboost,
+ cma,
+ cmaes,
+ colorlog,
+ distributed,
+ fakeredis,
+ google-cloud-storage,
+ lightgbm,
+ matplotlib,
+ mlflow,
+ moto,
+ numpy,
+ packaging,
+ pandas,
+ plotly,
+ pytest-xdist,
+ pytorch-lightning,
+ pyyaml,
+ redis,
+ scikit-learn,
+ scikit-optimize,
+ scipy,
+ setuptools,
+ shap,
+ sqlalchemy,
+ tensorflow,
+ torch,
+ torchaudio,
+ torchvision,
+ tqdm,
+ wandb,
+ wheel,
+ xgboost,
}:
buildPythonPackage rec {
@@ -114,12 +115,9 @@ buildPythonPackage rec {
pytest-xdist
pytestCheckHook
scipy
- ] ++ fakeredis.optional-dependencies.lua
- ++ passthru.optional-dependencies.optional;
+ ] ++ fakeredis.optional-dependencies.lua ++ passthru.optional-dependencies.optional;
- pytestFlagsArray = [
- "-m 'not integration'"
- ];
+ pytestFlagsArray = [ "-m 'not integration'" ];
disabledTestPaths = [
# require unpackaged kaleido and building it is a bit difficult
@@ -128,9 +126,7 @@ buildPythonPackage rec {
"tests/artifacts_tests/test_boto3.py"
];
- pythonImportsCheck = [
- "optuna"
- ];
+ pythonImportsCheck = [ "optuna" ];
meta = with lib; {
description = "A hyperparameter optimization framework";