diff --git a/pkgs/development/python-modules/sklearn-compat/default.nix b/pkgs/development/python-modules/sklearn-compat/default.nix new file mode 100644 index 000000000000..3fd17c6cade4 --- /dev/null +++ b/pkgs/development/python-modules/sklearn-compat/default.nix @@ -0,0 +1,52 @@ +{ + lib, + buildPythonPackage, + fetchFromGitHub, + hatchling, + scikit-learn, + pandas, + pytestCheckHook, + pytest-cov, + pytest-xdist, + pytz, +}: + +buildPythonPackage rec { + pname = "sklearn-compat"; + version = "0.1.3"; + pyproject = true; + + src = fetchFromGitHub { + owner = "sklearn-compat"; + repo = "sklearn-compat"; + tag = version; + hash = "sha256-9Py5VKDej7xDMdWVujkS/ooxOkEcjPJNlBHkbe/rycE="; + }; + + build-system = [ + hatchling + ]; + + dependencies = [ + scikit-learn + ]; + + nativeCheckInputs = [ + pandas + pytestCheckHook + pytest-cov + pytest-xdist + pytz + ]; + + pythonImportsCheck = [ + "sklearn_compat" + ]; + + meta = { + description = "Ease multi-version support for scikit-learn compatible library"; + homepage = "https://github.com/sklearn-compat/sklearn-compat"; + license = lib.licenses.bsd3; + maintainers = with lib.maintainers; [ philipwilk ]; + }; +} diff --git a/pkgs/top-level/python-packages.nix b/pkgs/top-level/python-packages.nix index 7543f3ec1f8c..d2efbb0b1031 100644 --- a/pkgs/top-level/python-packages.nix +++ b/pkgs/top-level/python-packages.nix @@ -16428,6 +16428,8 @@ self: super: with self; { skl2onnx = callPackage ../development/python-modules/skl2onnx { }; + sklearn-compat = callPackage ../development/python-modules/sklearn-compat { }; + sklearn-deap = callPackage ../development/python-modules/sklearn-deap { }; skodaconnect = callPackage ../development/python-modules/skodaconnect { };