diff options
Diffstat (limited to 'pkgs/development/python-modules/dvclive/default.nix')
-rw-r--r-- | pkgs/development/python-modules/dvclive/default.nix | 105 |
1 files changed, 43 insertions, 62 deletions
diff --git a/pkgs/development/python-modules/dvclive/default.nix b/pkgs/development/python-modules/dvclive/default.nix index 7090433431a9e..33adbdf07a08b 100644 --- a/pkgs/development/python-modules/dvclive/default.nix +++ b/pkgs/development/python-modules/dvclive/default.nix @@ -1,38 +1,39 @@ -{ lib -, buildPythonPackage -, datasets -, dvc -, dvc-render -, dvc-studio-client -, fastai -, fetchFromGitHub -, funcy -, gto -, jsonargparse -, lightgbm -, lightning -, matplotlib -, mmcv -, numpy -, optuna -, pandas -, pillow -, psutil -, pynvml -, pythonOlder -, ruamel-yaml -, scikit-learn -, scmrepo -, setuptools-scm -, tensorflow -, torch -, transformers -, xgboost +{ + lib, + buildPythonPackage, + datasets, + dvc, + dvc-render, + dvc-studio-client, + fastai, + fetchFromGitHub, + funcy, + gto, + jsonargparse, + lightgbm, + lightning, + matplotlib, + mmcv, + numpy, + optuna, + pandas, + pillow, + psutil, + pynvml, + pythonOlder, + ruamel-yaml, + scikit-learn, + scmrepo, + setuptools-scm, + tensorflow, + torch, + transformers, + xgboost, }: buildPythonPackage rec { pname = "dvclive"; - version = "3.45.0"; + version = "3.46.0"; pyproject = true; disabled = pythonOlder "3.9"; @@ -41,12 +42,10 @@ buildPythonPackage rec { owner = "iterative"; repo = "dvclive"; rev = "refs/tags/${version}"; - hash = "sha256-76PjBPr3m1y5osW024dkhuvuXl2FiZ5oxlJ1Ucq8msI="; + hash = "sha256-yIViKlkCdoG2vSZdScL38fZd9musLRKzBd9wSR6lJdk="; }; - build-system = [ - setuptools-scm - ]; + build-system = [ setuptools-scm ]; dependencies = [ dvc @@ -81,29 +80,17 @@ buildPythonPackage rec { numpy pillow ]; - sklearn = [ - scikit-learn - ]; + sklearn = [ scikit-learn ]; plots = [ pandas scikit-learn numpy ]; - markdown = [ - matplotlib - ]; - mmcv = [ - mmcv - ]; - tf = [ - tensorflow - ]; - xgb = [ - xgboost - ]; - lgbm = [ - lightgbm - ]; + markdown = [ matplotlib ]; + mmcv = [ mmcv ]; + tf = [ tensorflow ]; + xgb = [ xgboost ]; + lgbm = [ lightgbm ]; huggingface = [ datasets transformers @@ -111,25 +98,19 @@ buildPythonPackage rec { # catalyst = [ # catalyst # ]; - fastai = [ - fastai - ]; + fastai = [ fastai ]; lightning = [ lightning torch jsonargparse ] ++ jsonargparse.optional-dependencies.signatures; - optuna = [ - optuna - ]; + optuna = [ optuna ]; }; # Circular dependency with dvc doCheck = false; - pythonImportsCheck = [ - "dvclive" - ]; + pythonImportsCheck = [ "dvclive" ]; meta = with lib; { description = "Library for logging machine learning metrics and other metadata in simple file formats"; |