Something went wrong. Try again.
What happens if I push all of nixpkgs
Something went wrong. Try again.
2.2 kB · 140 lines
Nix
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141{ 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 { pname = "optuna"; version = "3.5.0"; format = "pyproject";
disabled = pythonOlder "3.7";
src = fetchFromGitHub { owner = "optuna"; repo = "optuna"; rev = "refs/tags/v${version}"; hash = "sha256-lNurMkZZKpKXXQoqhuRUv8LCbdSi1ryj3kYYioYZPF0="; };
nativeBuildInputs = [ setuptools wheel ];
propagatedBuildInputs = [ alembic colorlog numpy packaging sqlalchemy tqdm pyyaml ];
passthru.optional-dependencies = { integration = [ botorch catboost cma distributed lightgbm mlflow pandas # pytorch-ignite pytorch-lightning scikit-learn scikit-optimize 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 ++ passthru.optional-dependencies.optional;
pytestFlagsArray = [ "-m 'not integration'" ];
disabledTestPaths = [ # require unpackaged kaleido and building it is a bit difficult "tests/visualization_tests" ];
pythonImportsCheck = [ "optuna" ];
meta = with lib; { description = "A hyperparameter optimization framework"; homepage = "https://optuna.org/"; changelog = "https://github.com/optuna/optuna/releases/tag/${src.rev}"; license = licenses.mit; maintainers = with maintainers; [ natsukium ]; };}