diff --git a/pkgs/development/python-modules/dvclive/default.nix b/pkgs/development/python-modules/dvclive/default.nix index 33adbdf07a08..fd78c6ca873d 100644 --- a/pkgs/development/python-modules/dvclive/default.nix +++ b/pkgs/development/python-modules/dvclive/default.nix @@ -1,34 +1,33 @@ -{ - 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 { @@ -45,7 +44,9 @@ buildPythonPackage rec { hash = "sha256-yIViKlkCdoG2vSZdScL38fZd9musLRKzBd9wSR6lJdk="; }; - build-system = [ setuptools-scm ]; + build-system = [ + setuptools-scm + ]; dependencies = [ dvc @@ -80,17 +81,29 @@ 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 @@ -98,19 +111,25 @@ 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";