Something went wrong. Try again.
What happens if I push all of nixpkgs
Something went wrong. Try again.
2.3 kB · 138 lines
Nix
at gcc-offload
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139{ 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, scipy, setuptools, shap, sqlalchemy, tensorflow, torch, torchaudio, torchvision, tqdm, wandb, xgboost,}:
buildPythonPackage rec { pname = "optuna"; version = "4.1.0"; pyproject = true;
disabled = pythonOlder "3.8";
src = fetchFromGitHub { owner = "optuna"; repo = "optuna"; rev = "refs/tags/v${version}"; hash = "sha256-wIgYExxJEWFxEadBuCsxEIcW2/J6EVybW1jp83gIMjY="; };
build-system = [ setuptools ];
dependencies = [ alembic colorlog numpy packaging sqlalchemy tqdm pyyaml ];
optional-dependencies = { integration = [ botorch catboost cma distributed lightgbm mlflow pandas # pytorch-ignite pytorch-lightning scikit-learn shap tensorflow torch torchaudio torchvision wandb xgboost ]; optional = [ boto3 botorch cmaes google-cloud-storage matplotlib pandas plotly redis scikit-learn ]; };
preCheck = '' export PATH=$out/bin:$PATH '';
nativeCheckInputs = [ fakeredis moto pytest-xdist pytestCheckHook scipy ] ++ fakeredis.optional-dependencies.lua ++ optional-dependencies.optional;
pytestFlagsArray = [ "-m 'not integration'" ];
disabledTestPaths = [ # require unpackaged kaleido and building it is a bit difficult "tests/visualization_tests" # ImportError: cannot import name 'mock_s3' from 'moto' "tests/artifacts_tests/test_boto3.py" ];
pythonImportsCheck = [ "optuna" ];
meta = with lib; { description = "Hyperparameter optimization framework"; homepage = "https://optuna.org/"; changelog = "https://github.com/optuna/optuna/releases/tag/${version}"; license = licenses.mit; maintainers = with maintainers; [ natsukium ]; mainProgram = "optuna"; };}