Commit 595185b2 authored by Fabian Affolter's avatar Fabian Affolter
Browse files
parent aa724f81
Loading
Loading
Loading
Loading
+87 −3
Original line number Diff line number Diff line
{ lib
, buildPythonPackage
, datasets
, dvc
, dvc-render
, dvc-studio-client
, fastai
, fetchFromGitHub
, funcy
, gto
, jsonargparse
, lightgbm
, lightning
, matplotlib
, mmcv
, numpy
, optuna
, pandas
, pillow
, pytestCheckHook
, pythonOlder
, ruamel-yaml
, scikit-learn
, scmrepo
, setuptools-scm
, tabulate
, tensorflow
, torch
, transformers
, xgboost
}:

buildPythonPackage rec {
  pname = "dvclive";
  version = "3.5.1";
  format = "pyproject";
  version = "3.41.1";
  pyproject = true;

  disabled = pythonOlder "3.8";

@@ -23,7 +41,7 @@ buildPythonPackage rec {
    owner = "iterative";
    repo = pname;
    rev = "refs/tags/${version}";
    hash = "sha256-QsA8HZ6wIWKvtQ+f3nyRKKZRNJS56eZ1sKw+KNHxfXc=";
    hash = "sha256-PbgazRK3+CoJISh1ZXGjxDfbKHY/XqSvVrkpycvPi7c=";
  };

  nativeBuildInputs = [
@@ -32,12 +50,78 @@ buildPythonPackage rec {

  propagatedBuildInputs = [
    dvc
    dvc-render
    dvc-studio-client
    funcy
    gto
    ruamel-yaml
    scmrepo
  ];

  passthru.optional-dependencies = {
    all = [
      jsonargparse
      lightgbm
      lightning
      matplotlib
      mmcv
      numpy
      optuna
      pandas
      pillow
      scikit-learn
      tensorflow
      torch
      transformers
      xgboost
    ] ++ jsonargparse.optional-dependencies.signatures;
    image = [
      numpy
      pillow
    ];
    sklearn = [
      scikit-learn
    ];
    plots = [
      pandas
      scikit-learn
      numpy
    ];
    markdown = [
      matplotlib
    ];
    mmcv = [
      mmcv
    ];
    tf = [
      tensorflow
    ];
    xgb = [
      xgboost
    ];
    lgbm = [
      lightgbm
    ];
    huggingface = [
      datasets
      transformers
    ];
    # catalyst = [
    #   catalyst
    # ];
    fastai = [
      fastai
    ];
    lightning = [
      lightning
      torch
      jsonargparse
    ] ++ jsonargparse.optional-dependencies.signatures;
    optuna = [
      optuna
    ];
  };

  # Circular dependency with dvc
  doCheck = false;