Something went wrong. Try again.
What happens if I push all of nixpkgs
Something went wrong. Try again.
3.3 kB · 187 lines
Nix
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188{ lib, buildPythonPackage, fetchFromGitHub, pythonOlder, pythonRelaxDepsHook, deprecated, rich, backoff, packaging, pydantic, typer, tqdm, wrapt, numpy, httpx, pandas, monotonic# optional-dependencies, fastapi, opensearch-py, elasticsearch8, uvicorn, smart-open, brotli-asgi, alembic, sqlalchemy, greenlet, aiosqlite, luqum, scikit-learn, aiofiles, pyyaml, python-multipart, python-jose, passlib, psutil# , segment-analytics-python, asyncpg, psycopg2, schedule, prodict, cleanlab, datasets, huggingface-hub# , flair, faiss, flyingsquid, pgmpy, plotly, snorkel, spacy, transformers, evaluate, seqeval# , setfit# , span_marker, openai, peft# test dependencies, pytestCheckHook, pytest-cov, pytest-mock, pytest-asyncio, factory-boy}:let pname = "argilla"; version = "1.22.0"; optional-dependencies = { server = [ fastapi opensearch-py elasticsearch8 uvicorn smart-open brotli-asgi alembic sqlalchemy greenlet aiosqlite luqum scikit-learn aiofiles pyyaml python-multipart python-jose passlib psutil # segment-analytics-python ] ++ elasticsearch8.optional-dependencies.async ++ uvicorn.optional-dependencies.standard ++ python-jose.optional-dependencies.cryptography ++ passlib.optional-dependencies.bcrypt; postgresql = [ asyncpg psycopg2 ]; listeners = [ schedule prodict ]; integrations = [ pyyaml cleanlab datasets huggingface-hub # flair faiss flyingsquid pgmpy plotly snorkel spacy transformers evaluate seqeval # setfit # span_marker openai peft ] ++ transformers.optional-dependencies.torch; };inbuildPythonPackage { inherit pname version; format = "setuptools";
disabled = pythonOlder "3.8";
src = fetchFromGitHub { owner = "argilla-io"; repo = pname; rev = "refs/tags/v${version}"; hash = "sha256-aPq/ZPewQId5OZq5v+TgvWWBS03QDftYLEwhzNLzrFQ="; };
pythonRelaxDeps = [ "typer" "rich" "numpy" ];
nativeBuildInputs = [ pythonRelaxDepsHook ];
propagatedBuildInputs = [ httpx deprecated packaging pandas pydantic wrapt numpy tqdm backoff monotonic rich typer ];
# still quite a bit of optional dependencies missing doCheck = false;
preCheck = '' export HOME=$(mktemp -d) '';
nativeCheckInputs = [ pytestCheckHook pytest-cov pytest-mock pytest-asyncio factory-boy ] ++ optional-dependencies.server ++ optional-dependencies.postgresql ++ optional-dependencies.listeners ++ optional-dependencies.integrations;
pytestFlagsArray = [ "--ignore=tests/server/datasets/test_dao.py" ];
passthru.optional-dependencies = optional-dependencies;
meta = with lib; { description = "Argilla: the open-source data curation platform for LLMs"; homepage = "https://github.com/argilla-io/argilla"; changelog = "https://github.com/argilla-io/argilla/releases/tag/v${version}"; license = licenses.asl20; maintainers = with maintainers; [ happysalada ]; };}