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

   1 { lib
   2 , buildPythonPackage
   3 , fetchFromGitHub
   4 , cloudpickle
   5 , scikit-learn
   6 , scikitimage
   7 , packaging
   8 , psutil
   9 , py-deprecate
  10 , torch
  11 , pytestCheckHook
  12 , torchmetrics
  13 , pytorch-lightning
  14 }:
  15
  16 let
  17   pname = "torchmetrics";
  18   version = "0.9.3";
  19 in
  20 buildPythonPackage {
  21   inherit pname version;
  22
  23   src = fetchFromGitHub {
  24     owner = "PyTorchLightning";
  25     repo = "metrics";
  26     rev = "refs/tags/v${version}";
  27     hash = "sha256-L2p8UftRkuBuRJX4V5+OYkJeJ5pCK3MvfA1OvSfgglY=";
  28   };
  29
  30   propagatedBuildInputs = [
  31     packaging
  32     py-deprecate
  33   ];
  34
  35   # Let the user bring their own instance
  36   buildInputs = [
  37     torch
  38   ];
  39
  40   checkInputs = [
  41     pytorch-lightning
  42     scikit-learn
  43     scikitimage
  44     cloudpickle
  45     psutil
  46     pytestCheckHook
  47   ];
  48
  49   # A cyclic dependency in: integrations/test_lightning.py
  50   doCheck = false;
  51   passthru.tests.check = torchmetrics.overridePythonAttrs (_: {
  52     doCheck = true;
  53   });
  54
  55   disabledTestPaths = [
  56     # These require too many "leftpad-level" dependencies
  57     "tests/text"
  58     "tests/audio"
  59     "tests/image"
  60
  61     # A few non-deterministic things like test_check_compute_groups_is_faster
  62     "tests/bases/test_collections.py"
  63   ];
  64
  65   pythonImportsCheck = [
  66     "torchmetrics"
  67   ];
  68
  69   meta = with lib; {
  70     description = "Machine learning metrics for distributed, scalable PyTorch applications (used in pytorch-lightning)";
  71     homepage = "https://torchmetrics.readthedocs.io";
  72     license = licenses.asl20;
  73     maintainers = with maintainers; [
  74       SomeoneSerge
  75     ];
  76   };
  77 }
  78