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

   1 {
   2   lib,
   3   buildPythonPackage,
   4   datasets,
   5   dvc,
   6   dvc-render,
   7   dvc-studio-client,
   8   fastai,
   9   fetchFromGitHub,
  10   funcy,
  11   gto,
  12   jsonargparse,
  13   lightgbm,
  14   lightning,
  15   matplotlib,
  16   mmcv,
  17   numpy,
  18   optuna,
  19   pandas,
  20   pillow,
  21   psutil,
  22   pynvml,
  23   pythonOlder,
  24   ruamel-yaml,
  25   scikit-learn,
  26   scmrepo,
  27   setuptools-scm,
  28   tensorflow,
  29   torch,
  30   transformers,
  31   xgboost,
  32 }:
  33
  34 buildPythonPackage rec {
  35   pname = "dvclive";
  36   version = "3.48.0";
  37   pyproject = true;
  38
  39   disabled = pythonOlder "3.9";
  40
  41   src = fetchFromGitHub {
  42     owner = "iterative";
  43     repo = "dvclive";
  44     rev = "refs/tags/${version}";
  45     hash = "sha256-WIVRpJD7B6OI7ZfdHT+DunRRiaxHhri5Ge/B1BQ1kJY=";
  46   };
  47
  48   build-system = [ setuptools-scm ];
  49
  50   dependencies = [
  51     dvc
  52     dvc-render
  53     dvc-studio-client
  54     funcy
  55     gto
  56     ruamel-yaml
  57     scmrepo
  58     psutil
  59     pynvml
  60   ];
  61
  62   optional-dependencies = {
  63     all = [
  64       jsonargparse
  65       lightgbm
  66       lightning
  67       matplotlib
  68       mmcv
  69       numpy
  70       optuna
  71       pandas
  72       pillow
  73       scikit-learn
  74       tensorflow
  75       torch
  76       transformers
  77       xgboost
  78     ] ++ jsonargparse.optional-dependencies.signatures;
  79     image = [
  80       numpy
  81       pillow
  82     ];
  83     sklearn = [ scikit-learn ];
  84     plots = [
  85       pandas
  86       scikit-learn
  87       numpy
  88     ];
  89     markdown = [ matplotlib ];
  90     mmcv = [ mmcv ];
  91     tf = [ tensorflow ];
  92     xgb = [ xgboost ];
  93     lgbm = [ lightgbm ];
  94     huggingface = [
  95       datasets
  96       transformers
  97     ];
  98     # catalyst = [
  99     #   catalyst
 100     # ];
 101     fastai = [ fastai ];
 102     lightning = [
 103       lightning
 104       torch
 105       jsonargparse
 106     ] ++ jsonargparse.optional-dependencies.signatures;
 107     optuna = [ optuna ];
 108   };
 109
 110   # Circular dependency with dvc
 111   doCheck = false;
 112
 113   pythonImportsCheck = [ "dvclive" ];
 114
 115   meta = with lib; {
 116     description = "Library for logging machine learning metrics and other metadata in simple file formats";
 117     homepage = "https://github.com/iterative/dvclive";
 118     changelog = "https://github.com/iterative/dvclive/releases/tag/${version}";
 119     license = licenses.asl20;
 120     maintainers = with maintainers; [ fab ];
 121   };
 122 }