tjh.dev
1{ lib
2, buildPythonPackage
3, fetchFromGitHub
4, pythonOlder
5, scikit-learn
6, termcolor
7, tqdm
8, pandas
9# test dependencies
10, tensorflow
11, torch
12, datasets
13, torchvision
14, keras
15, fasttext
16}:
17let
18 pname = "cleanlab";
19 version = "2.4.0";
20in
21buildPythonPackage {
22 inherit pname version;
23 format = "setuptools";
24
25 disabled = pythonOlder "3.8";
26
27 src = fetchFromGitHub {
28 owner = "cleanlab";
29 repo = pname;
30 rev = "v${version}";
31 hash = "sha256-XFrjjBJA0OQEAspnQQiSIW4td0USJDXTp9C/91mobp8=";
32 };
33
34 # postPatch = ''
35 # substituteInPlace pyproject.toml \
36 # --replace '"rich <= 13.0.1"' '"rich"' \
37 # --replace '"numpy < 1.24.0"' '"numpy"'
38 # '';
39
40 propagatedBuildInputs = [
41 scikit-learn
42 termcolor
43 tqdm
44 pandas
45 ];
46
47 nativeCheckInputs = [
48 tensorflow
49 torch
50 datasets
51 torchvision
52 keras
53 fasttext
54 ];
55
56 meta = with lib; {
57 description = "The standard data-centric AI package for data quality and machine learning with messy, real-world data and labels.";
58 homepage = "https://github.com/cleanlab/cleanlab";
59 changelog = "https://github.com/cleanlab/cleanlab/releases/tag/v${version}";
60 license = licenses.agpl3Only;
61 maintainers = with maintainers; [ happysalada ];
62 };
63}