pkgs/development/python-modules/skops/default.nix

   1 {
   2   lib,
   3   buildPythonPackage,
   4   fetchFromGitHub,
   5   pythonOlder,
   6   setuptools,
   7   pytestCheckHook,
   8   pytest-cov-stub,
   9   huggingface-hub,
  10   matplotlib,
  11   pandas,
  12   scikit-learn,
  13   stdenv,
  14   streamlit,
  15   tabulate,
  16 }:
  17
  18 buildPythonPackage rec {
  19   pname = "skops";
  20   version = "0.10";
  21   pyproject = true;
  22
  23   disabled = pythonOlder "3.9";
  24
  25   src = fetchFromGitHub {
  26     owner = "skops-dev";
  27     repo = "skops";
  28     rev = "refs/tags/v${version}";
  29     hash = "sha256-2uX5sGVdTnZEbl0VXI8E7h1pQYQVbpQeUKUchCZpgg4=";
  30   };
  31
  32   build-system = [ setuptools ];
  33
  34   dependencies = [
  35     huggingface-hub
  36     scikit-learn
  37     tabulate
  38   ];
  39
  40   nativeCheckInputs = [
  41     matplotlib
  42     pandas
  43     pytestCheckHook
  44     pytest-cov-stub
  45     streamlit
  46   ];
  47   pytestFlagsArray = [ "skops" ];
  48   disabledTestPaths =
  49     [
  50       # try to download data from Huggingface Hub:
  51       "skops/hub_utils/tests"
  52       "skops/card/tests"
  53       # minor output formatting issue
  54       "skops/card/_model_card.py"
  55     ]
  56     ++ lib.optionals stdenv.hostPlatform.isDarwin [
  57       # Segfaults on darwin
  58       "skops/io/tests/test_persist.py"
  59     ];
  60
  61   pythonImportsCheck = [ "skops" ];
  62
  63   meta = {
  64     description = "Library for saving/loading, sharing, and deploying scikit-learn based models";
  65     mainProgram = "skops";
  66     homepage = "https://skops.readthedocs.io/en/stable";
  67     changelog = "https://github.com/skops-dev/skops/releases/tag/v${version}";
  68     license = lib.licenses.mit;
  69     maintainers = [ lib.maintainers.bcdarwin ];
  70   };
  71 }