Something went wrong. Try again.
What happens if I push all of nixpkgs
Something went wrong. Try again.
2.7 kB · 118 lines
Nix
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119{ lib, buildPythonPackage, fetchFromGitHub,
# build-system setuptools,
# dependencies numpy, scikit-learn, termcolor, tqdm, pandas,
# tests cleanvision, datasets, fasttext, hypothesis, keras, matplotlib, pytestCheckHook, pytest-lazy-fixture, skorch, tensorflow, torch, torchvision, wget, pythonAtLeast,}:
buildPythonPackage rec { pname = "cleanlab"; version = "2.7.1"; pyproject = true;
src = fetchFromGitHub { owner = "cleanlab"; repo = "cleanlab"; tag = "v${version}"; hash = "sha256-KzVqBOLTxxkgvoGPYMeYb7zMuG8VwQwX6SYR/FUhfBw="; };
build-system = [ setuptools ];
pythonRelaxDeps = [ "numpy" ];
dependencies = [ numpy scikit-learn termcolor tqdm pandas ];
# This is ONLY turned off when we have testing enabled. # The reason we do this is because of duplicate packages in the enclosure # when using the packages in nativeCheckInputs. # Affected packages: grpcio protobuf tensorboard tensorboard-plugin-profile catchConflicts = (!doCheck); doCheck = true;
nativeCheckInputs = [ cleanvision datasets fasttext hypothesis keras matplotlib pytestCheckHook pytest-lazy-fixture skorch tensorflow torch torchvision wget ];
disabledTests = [ # Incorrect snapshots (AssertionError) "test_color_sentence"
# Requires the datasets we prevent from downloading "test_create_imagelab"
# Non-trivial numpy2 incompatibilities # assert np.float64(0.492) == 0.491 "test_duplicate_points_have_similar_scores" # AssertionError: assert 'Annotators [1] did not label any examples.' "test_label_quality_scores_multiannotator" ] ++ lib.optionals (pythonAtLeast "3.12") [ # AttributeError: 'called_once_with' is not a valid assertion. # Use a spec for the mock if 'called_once_with' is meant to be an attribute.. # Did you mean: 'assert_called_once_with'? "test_custom_issue_manager_not_registered" ];
disabledTestPaths = [ # Requires internet "tests/test_dataset.py" # Requires the datasets we just prevented from downloading "tests/datalab/test_cleanvision_integration.py" # Fails because of issues with the keras derivation "tests/test_frameworks.py" ];
meta = { description = "Standard data-centric AI package for data quality and machine learning with messy, real-world data and labels"; homepage = "https://github.com/cleanlab/cleanlab"; changelog = "https://github.com/cleanlab/cleanlab/releases/tag/v${version}"; license = lib.licenses.agpl3Only; maintainers = with lib.maintainers; [ happysalada ]; };}