about summary refs log tree commit diff
path: root/pkgs/development/python-modules/dvclive/default.nix
diff options
context:
space:
mode:
Diffstat (limited to 'pkgs/development/python-modules/dvclive/default.nix')
-rw-r--r--pkgs/development/python-modules/dvclive/default.nix105
1 files changed, 62 insertions, 43 deletions
diff --git a/pkgs/development/python-modules/dvclive/default.nix b/pkgs/development/python-modules/dvclive/default.nix
index 33adbdf07a08b..7090433431a9e 100644
--- a/pkgs/development/python-modules/dvclive/default.nix
+++ b/pkgs/development/python-modules/dvclive/default.nix
@@ -1,39 +1,38 @@
-{
-  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.46.0";
+  version = "3.45.0";
   pyproject = true;
 
   disabled = pythonOlder "3.9";
@@ -42,10 +41,12 @@ buildPythonPackage rec {
     owner = "iterative";
     repo = "dvclive";
     rev = "refs/tags/${version}";
-    hash = "sha256-yIViKlkCdoG2vSZdScL38fZd9musLRKzBd9wSR6lJdk=";
+    hash = "sha256-76PjBPr3m1y5osW024dkhuvuXl2FiZ5oxlJ1Ucq8msI=";
   };
 
-  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";