Commit 342b93ab authored by Gaetan Lepage's avatar Gaetan Lepage
Browse files
parent 3ce8806f
Loading
Loading
Loading
Loading
+24 −18
Original line number Diff line number Diff line
{
  lib,
  buildPythonPackage,
  datasets,
  fetchFromGitHub,

  # build-system
  setuptools-scm,

  # dependencies
  dvc,
  dvc-render,
  dvc-studio-client,
  fastai,
  fetchFromGitHub,
  funcy,
  gto,
  psutil,
  pynvml,
  ruamel-yaml,
  scmrepo,

  # optional-dependencies
  # all
  jsonargparse,
  lightgbm,
  lightning,
@@ -18,31 +28,27 @@
  optuna,
  pandas,
  pillow,
  psutil,
  pynvml,
  pythonOlder,
  ruamel-yaml,
  scikit-learn,
  scmrepo,
  setuptools-scm,
  tensorflow,
  torch,
  transformers,
  xgboost,
  # huggingface
  datasets,
  # fastai
  fastai,
}:

buildPythonPackage rec {
  pname = "dvclive";
  version = "3.48.4";
  version = "3.48.5";
  pyproject = true;

  disabled = pythonOlder "3.9";

  src = fetchFromGitHub {
    owner = "iterative";
    repo = "dvclive";
    tag = version;
    hash = "sha256-fs0NbsjpGaR4cXE/ajfeF6ZJ3fc7K28KDJ0LIJ/vsdg=";
    hash = "sha256-ucMtYHDwdpyYnnC7QCn5T6gCS8SarohKh6lxFXtPXgc=";
  };

  build-system = [ setuptools-scm ];
@@ -53,10 +59,10 @@ buildPythonPackage rec {
    dvc-studio-client
    funcy
    gto
    ruamel-yaml
    scmrepo
    psutil
    pynvml
    ruamel-yaml
    scmrepo
  ];

  optional-dependencies = {
@@ -114,11 +120,11 @@ buildPythonPackage rec {

  pythonImportsCheck = [ "dvclive" ];

  meta = with lib; {
  meta = {
    description = "Library for logging machine learning metrics and other metadata in simple file formats";
    homepage = "https://github.com/iterative/dvclive";
    changelog = "https://github.com/iterative/dvclive/releases/tag/${src.tag}";
    license = licenses.asl20;
    maintainers = with maintainers; [ fab ];
    license = lib.licenses.asl20;
    maintainers = with lib.maintainers; [ fab ];
  };
}