diff options
author | Connor Baker <connor.baker@tweag.io> | 2024-05-26 23:12:18 -0400 |
---|---|---|
committer | GitHub <noreply@github.com> | 2024-05-26 23:12:18 -0400 |
commit | 2663954f6770859f97b2326c9bbdc75411067677 (patch) | |
tree | f0764f1edf58a4c7e9eb804d476d3add680b62bf | |
parent | a23b799334c55878543467fd8cd27e04ae6a0635 (diff) | |
parent | 7568ee06c9f8db787588c2168895301a5db4cded (diff) |
Merge pull request #314707 from ConnorBaker/fix/python3Packages.apricot-select
python3Packages.apricot-select: fixup dependency categories and disable tests
-rw-r--r-- | pkgs/development/python-modules/apricot-select/default.nix | 20 |
1 files changed, 18 insertions, 2 deletions
diff --git a/pkgs/development/python-modules/apricot-select/default.nix b/pkgs/development/python-modules/apricot-select/default.nix index b50707e96e6fa..f05e9b8e5e95f 100644 --- a/pkgs/development/python-modules/apricot-select/default.nix +++ b/pkgs/development/python-modules/apricot-select/default.nix @@ -2,6 +2,7 @@ lib, buildPythonPackage, fetchFromGitHub, + apricot-select, numba, numpy, pynose, @@ -37,15 +38,15 @@ buildPythonPackage rec { dependencies = [ numba numpy + scikit-learn scipy + torchvision tqdm ]; nativeCheckInputs = [ pynose pytestCheckHook - scikit-learn - torchvision ]; pythonImportsCheck = [ "apricot" ]; @@ -56,6 +57,21 @@ buildPythonPackage rec { "tests/test_optimizers/test_knapsack_feature_based.py" ]; + # NOTE: These tests seem to be flaky. + disabledTests = [ + "test_digits_modular" + "test_digits_modular_object" + "test_digits_modular_sparse" + "test_digits_sqrt_modular" + "test_digits_sqrt_modular_object" + "test_digits_sqrt_modular_sparse" + ]; + + # NOTE: Tests are disabled by default because they can run for hours and timeout on Hydra. + doCheck = false; + + passthru.tests.check = apricot-select.overridePythonAttrs { doCheck = true; }; + meta = with lib; { description = "Module for submodular optimization for the purpose of selecting subsets of massive data sets"; homepage = "https://github.com/jmschrei/apricot"; |