From 343878138d6212f20b7a6c1572638f18e694a6dc Mon Sep 17 00:00:00 2001
From: Gaetan Lepage <gaetan@glepage.com>
Date: Tue, 28 May 2024 14:06:53 +0200
Subject: [PATCH] python311Packages.cleanlab: 2.6.1 -> 2.6.5

Diff: https://github.com/cleanlab/cleanlab/compare/refs/tags/v2.6.1...v2.6.5

Changelog: https://github.com/cleanlab/cleanlab/releases/tag/v2.6.5
---
 .../python-modules/cleanlab/default.nix       | 51 ++++++++++---------
 1 file changed, 28 insertions(+), 23 deletions(-)

diff --git a/pkgs/development/python-modules/cleanlab/default.nix b/pkgs/development/python-modules/cleanlab/default.nix
index e90989967b11..1d42c24609f8 100644
--- a/pkgs/development/python-modules/cleanlab/default.nix
+++ b/pkgs/development/python-modules/cleanlab/default.nix
@@ -3,42 +3,45 @@
   buildPythonPackage,
   fetchFromGitHub,
   pythonOlder,
+  numpy,
   scikit-learn,
   termcolor,
   tqdm,
   pandas,
   setuptools,
   # test dependencies
-  pytestCheckHook,
-  pytest-lazy-fixture,
-  tensorflow,
-  torch,
   datasets,
-  torchvision,
-  keras,
   fasttext,
   hypothesis,
-  wget,
+  keras,
   matplotlib,
+  pytestCheckHook,
+  pytest-lazy-fixture,
   skorch,
+  tensorflow,
+  torch,
+  torchvision,
+  wget,
 }:
 
 buildPythonPackage rec {
   pname = "cleanlab";
-  version = "2.6.1";
+  version = "2.6.5";
   pyproject = true;
-  disabled = pythonOlder "3.7";
+
+  disabled = pythonOlder "3.8";
 
   src = fetchFromGitHub {
     owner = "cleanlab";
-    repo = pname;
+    repo = "cleanlab";
     rev = "refs/tags/v${version}";
-    hash = "sha256-+uJtm/t6Ri25V/9N/2fcOgCOBaBy8PrsM/tO1uX7FEY=";
+    hash = "sha256-wehvGh27Ey1YK+eWTjT6jRwa7yqPpx3P0HUNePoljpw=";
   };
 
-  nativeBuildInputs = [ setuptools ];
+  build-system = [ setuptools ];
 
-  propagatedBuildInputs = [
+  dependencies = [
+    numpy
     scikit-learn
     termcolor
     tqdm
@@ -53,18 +56,18 @@ buildPythonPackage rec {
   doCheck = true;
 
   nativeCheckInputs = [
-    pytestCheckHook
-    pytest-lazy-fixture
-    tensorflow
-    torch
     datasets
-    torchvision
-    keras
     fasttext
     hypothesis
-    wget
+    keras
     matplotlib
+    pytestCheckHook
+    pytest-lazy-fixture
     skorch
+    tensorflow
+    torch
+    torchvision
+    wget
   ];
 
   disabledTests = [
@@ -77,13 +80,15 @@ buildPythonPackage rec {
     "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 = with lib; {
+  meta = {
     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 ];
+    license = lib.licenses.agpl3Only;
+    maintainers = with lib.maintainers; [ happysalada ];
   };
 }