flake/packages/hf-doc-builder/default.nix
Max 079f83ba45 Major Upgrade/Refactor
- KoboldAI is no longer upstream maintained, so is now deprecated in
  nixified.ai

- InvokeAI: v2.3.1.post2 -> v3.3.0post3
- textgen: init at v1.7
- treewide: update flake inputs including nixpkgs
- treewide: add a bunch of new dependencies and upgrade old ones
2023-10-19 22:13:37 +01:00

88 lines
1.3 KiB
Nix

{ lib
, buildPythonPackage
, fetchPypi
, git-python
, gql
, nbformat
, packaging
, pyyaml
, requests
, tqdm
, black
, flake8
, isort
, pytest
, pytest-xdist
, tokenizers
, torch
, transformers
}:
buildPythonPackage rec {
pname = "hf-doc-builder";
version = "0.4.0";
format = "pyproject";
src = fetchPypi {
inherit pname version;
hash = "sha256-I8AnfIa9IK5lXAn8oHJzyvJT51VNFFmnKIWhhhYhVI0=";
};
propagatedBuildInputs = [
git-python
gql
nbformat
packaging
pyyaml
requests
tqdm
];
passthru.optional-dependencies = {
all = [
black
flake8
isort
pytest
pytest-xdist
tokenizers
torch
transformers
];
dev = [
black
flake8
isort
pytest
pytest-xdist
tokenizers
torch
transformers
];
quality = [
black
flake8
isort
];
testing = [
pytest
pytest-xdist
tokenizers
torch
transformers
];
transformers = [
transformers
];
};
pythonImportsCheck = [ "hf-doc-builder" ];
meta = with lib; {
description = "Doc building utility";
homepage = "https://github.com/huggingface/doc-builder";
license = licenses.asl20;
maintainers = with maintainers; [ jpetrucciani ];
};
}