diff --git a/pkgs/development/python-modules/cleanlab/default.nix b/pkgs/development/python-modules/cleanlab/default.nix new file mode 100644 index 000000000000..0ac79e306c17 --- /dev/null +++ b/pkgs/development/python-modules/cleanlab/default.nix @@ -0,0 +1,63 @@ +{ lib +, buildPythonPackage +, fetchFromGitHub +, pythonOlder +, scikit-learn +, termcolor +, tqdm +, pandas +# test dependencies +, tensorflow +, torch +, datasets +, torchvision +, keras +, fasttext +}: +let + pname = "cleanlab"; + version = "2.4.0"; +in +buildPythonPackage { + inherit pname version; + format = "setuptools"; + + disabled = pythonOlder "3.8"; + + src = fetchFromGitHub { + owner = "cleanlab"; + repo = pname; + rev = "v${version}"; + hash = "sha256-XFrjjBJA0OQEAspnQQiSIW4td0USJDXTp9C/91mobp8="; + }; + + # postPatch = '' + # substituteInPlace pyproject.toml \ + # --replace '"rich <= 13.0.1"' '"rich"' \ + # --replace '"numpy < 1.24.0"' '"numpy"' + # ''; + + propagatedBuildInputs = [ + scikit-learn + termcolor + tqdm + pandas + ]; + + nativeCheckInputs = [ + tensorflow + torch + datasets + torchvision + keras + fasttext + ]; + + meta = with lib; { + description = "The 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 = licenses.agpl3Only; + maintainers = with maintainers; [ happysalada ]; + }; +} diff --git a/pkgs/top-level/python-packages.nix b/pkgs/top-level/python-packages.nix index 10c288d67b0a..fd8f70965805 100644 --- a/pkgs/top-level/python-packages.nix +++ b/pkgs/top-level/python-packages.nix @@ -1863,6 +1863,8 @@ self: super: with self; { clean-fid = callPackage ../development/python-modules/clean-fid { }; + cleanlab = callPackage ../development/python-modules/cleanlab { }; + cleo = callPackage ../development/python-modules/cleo { }; clevercsv = callPackage ../development/python-modules/clevercsv { };