diff options
Diffstat (limited to 'pkgs/development/python-modules/optuna/default.nix')
| -rw-r--r-- | pkgs/development/python-modules/optuna/default.nix | 94 |
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"; |
