diff --git a/pkgs/development/python-modules/tqdm-multiprocess/default.nix b/pkgs/development/python-modules/tqdm-multiprocess/default.nix new file mode 100644 index 000000000000..812a694078d0 --- /dev/null +++ b/pkgs/development/python-modules/tqdm-multiprocess/default.nix @@ -0,0 +1,41 @@ +{ + lib, + buildPythonPackage, + fetchFromGitHub, + setuptools-scm, + tqdm, + colorama, +}: + +buildPythonPackage { + pname = "tqdm-multiprocess"; + version = "0.0.11"; + pyproject = true; + + src = fetchFromGitHub { + owner = "EleutherAI"; + repo = "tqdm-multiprocess"; + rev = "fccefc473595055bf3a5e74bcf8a75b3a9517638"; + hash = "sha256-nQeFPwF5OasOYrVs7kLG/Uz6pf1FKxar4ygggo8s4ZM="; + }; + + build-system = [ + setuptools-scm + ]; + + dependencies = [ + colorama + tqdm + ]; + + pythonImportsCheck = [ + "tqdm_multiprocess" + ]; + + meta = { + description = "Support multiple worker processes, each with multiple tqdm progress bars, displaying them cleanly through the main process"; + homepage = "https://github.com/EleutherAI/tqdm-multiprocess"; + license = [ lib.licenses.mit ]; + maintainers = [ lib.maintainers.booxter ]; + }; +} diff --git a/pkgs/top-level/python-packages.nix b/pkgs/top-level/python-packages.nix index bef5491866bc..9f56f2c03415 100644 --- a/pkgs/top-level/python-packages.nix +++ b/pkgs/top-level/python-packages.nix @@ -16401,6 +16401,8 @@ self: super: with self; { tqdm = callPackage ../development/python-modules/tqdm { }; + tqdm-multiprocess = callPackage ../development/python-modules/tqdm-multiprocess { }; + traceback2 = callPackage ../development/python-modules/traceback2 { }; tracerite = callPackage ../development/python-modules/tracerite { };