Something went wrong. Try again.
lol
Something went wrong. Try again.
3.2 kB · 193 lines
Nix
at master
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194{ lib, aiofiles, aiosqlite, alembic, asyncpg, backoff, brotli-asgi, buildPythonPackage, cleanlab, datasets, elasticsearch8, evaluate, factory-boy, faiss, fastapi, fetchFromGitHub, flyingsquid, greenlet, httpx, huggingface-hub, luqum, monotonic, numpy, openai, opensearch-py, packaging, pandas, passlib, pdm-backend, peft, pgmpy, pillow, plotly, prodict, psutil, psycopg2, pydantic, pytest-asyncio, pytest-mock, pytestCheckHook, python-jose, python-multipart, pyyaml, rich, schedule, scikit-learn, sentence-transformers, seqeval, smart-open, snorkel, spacy-transformers, spacy, sqlalchemy, standardwebhooks, tqdm, transformers, typer, uvicorn, wrapt, # , flair # , setfit # , spacy-huggingface-hub # , span_marker # , trl}:
buildPythonPackage rec { pname = "argilla"; version = "2.8.0"; pyproject = true;
src = fetchFromGitHub { owner = "argilla-io"; repo = "argilla"; tag = "v${version}"; hash = "sha256-8j7/Gtn4FnAZA3oIV7dLxKwNtigqB7AweHtQ/kzLwm4="; };
sourceRoot = "${src.name}/${pname}";
pythonRelaxDeps = [ "httpx" "numpy" "rich" "typer" "wrapt" ];
build-system = [ pdm-backend ];
dependencies = [ httpx datasets packaging pandas pydantic wrapt numpy tqdm pillow huggingface-hub monotonic rich typer standardwebhooks ];
optional-dependencies = { server = [ aiofiles aiosqlite alembic backoff brotli-asgi elasticsearch8 fastapi greenlet luqum opensearch-py passlib psutil python-jose python-multipart pyyaml scikit-learn smart-open sqlalchemy uvicorn ] ++ elasticsearch8.optional-dependencies.async ++ uvicorn.optional-dependencies.standard ++ python-jose.optional-dependencies.cryptography ++ passlib.optional-dependencies.bcrypt; postgresql = [ asyncpg psycopg2 ]; listeners = [ schedule prodict ]; integrations = [ cleanlab evaluate faiss flyingsquid openai peft pgmpy plotly pyyaml sentence-transformers seqeval snorkel spacy spacy-transformers transformers # flair # setfit # span_marker # trl # spacy-huggingface-hub ] ++ transformers.optional-dependencies.torch; };
# Still quite a bit of optional dependencies missing doCheck = false;
preCheck = '' export HOME=$(mktemp -d) '';
nativeCheckInputs = [ pytestCheckHook pytest-mock pytest-asyncio factory-boy ] ++ lib.concatAttrValues optional-dependencies;
disabledTestPaths = [ "tests/server/datasets/test_dao.py" ];
meta = { description = "Open-source data curation platform for LLMs"; homepage = "https://github.com/argilla-io/argilla"; changelog = "https://github.com/argilla-io/argilla/releases/tag/${src.tag}"; license = lib.licenses.asl20; maintainers = with lib.maintainers; [ happysalada ]; mainProgram = "argilla"; };}