Something went wrong. Try again.
lol
Something went wrong. Try again.
2.1 kB · 121 lines
Nix
at master
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122{ lib, accelerate, aiohttp, buildPythonPackage, fastapi, fetchFromGitHub, flask, numpy, pg8000, pillow, pydantic, pytestCheckHook, redis, requests, sentence-transformers, setuptools, sqlalchemy, sqlitedict, tenacity, tiktoken, torch, transformers, uvicorn, xxhash,}:
buildPythonPackage rec { pname = "manifest-ml"; version = "0.1.9"; pyproject = true;
src = fetchFromGitHub { owner = "HazyResearch"; repo = "manifest"; tag = "v${version}"; hash = "sha256-6m1XZOXzflBYyq9+PinbrW+zqvNGFN/aRDHH1b2Me5E="; };
__darwinAllowLocalNetworking = true;
pythonRelaxDeps = [ "pydantic" ];
build-system = [ setuptools ];
dependencies = [ numpy pydantic redis requests aiohttp sqlitedict tenacity tiktoken xxhash ];
optional-dependencies = { api = [ accelerate # deepspeed # diffusers flask sentence-transformers torch transformers ]; app = [ fastapi uvicorn ]; diffusers = [ pillow ]; gcp = [ pg8000 # cloud-sql-python-connector sqlalchemy ]; };
nativeCheckInputs = [ pytestCheckHook ] ++ lib.concatAttrValues optional-dependencies;
preCheck = '' export HOME=$TMPDIR '';
disabledTestPaths = [ # this file tries importing `deepspeed`, which is not yet packaged in nixpkgs "tests/test_huggingface_api.py" ];
disabledTests = [ # Tests require DB access "test_init" "test_key_get_and_set" "test_get" # Tests require network access "test_abatch_run" "test_batch_run" "test_retry_handling" "test_run_chat" "test_run" "test_score_run" # Test is time-sensitive "test_timing" ];
pythonImportsCheck = [ "manifest" ];
meta = { description = "Manifest for Prompting Foundation Models"; homepage = "https://github.com/HazyResearch/manifest"; changelog = "https://github.com/HazyResearch/manifest/releases/tag/v${version}"; license = lib.licenses.asl20; maintainers = with lib.maintainers; [ natsukium ]; };}