71 lines
1.1 KiB
Nix
71 lines
1.1 KiB
Nix
|
{
|
||
|
|
lib,
|
||
|
|
buildPythonPackage,
|
||
|
|
fetchFromGitHub,
|
||
|
|
setuptools,
|
||
|
|
setuptools-scm,
|
||
|
|
interegular,
|
||
|
|
cloudpickle,
|
||
|
|
datasets,
|
||
|
|
diskcache,
|
||
|
|
joblib,
|
||
|
|
jsonschema,
|
||
|
|
pyairports,
|
||
|
|
pycountry,
|
||
|
|
pydantic,
|
||
|
|
lark,
|
||
|
|
nest-asyncio,
|
||
|
|
numba,
|
||
|
|
scipy,
|
||
|
|
torch,
|
||
|
|
transformers,
|
||
|
|
}:
|
||
|
|
|
||
|
|
buildPythonPackage rec {
|
||
|
|
pname = "outlines";
|
||
|
|
version = "0.0.46";
|
||
|
|
pyproject = true;
|
||
|
|
|
||
|
|
src = fetchFromGitHub {
|
||
|
|
owner = "outlines-dev";
|
||
|
|
repo = pname;
|
||
|
|
rev = "refs/tags/${version}";
|
||
|
|
hash = "sha256-6VH9BcMRVRf2xvLcK3GNA1pGgAOs95UOlFQ6KxHXwKo=";
|
||
|
|
};
|
||
|
|
|
||
|
|
nativeBuildInputs = [
|
||
|
|
setuptools
|
||
|
|
setuptools-scm
|
||
|
|
];
|
||
|
|
|
||
|
|
propagatedBuildInputs = [
|
||
|
|
interegular
|
||
|
|
cloudpickle
|
||
|
|
datasets
|
||
|
|
diskcache
|
||
|
|
joblib
|
||
|
|
jsonschema
|
||
|
|
pydantic
|
||
|
|
lark
|
||
|
|
nest-asyncio
|
||
|
|
numba
|
||
|
|
scipy
|
||
|
|
torch
|
||
|
|
transformers
|
||
|
|
pycountry
|
||
|
|
pyairports
|
||
|
|
];
|
||
|
|
|
||
|
|
checkPhase = ''
|
||
|
|
export HOME=$(mktemp -d)
|
||
|
|
python3 -c 'import outlines'
|
||
|
|
'';
|
||
|
|
|
||
|
|
meta = with lib; {
|
||
|
|
description = "Structured text generation";
|
||
|
|
homepage = "https://github.com/outlines-dev/outlines";
|
||
|
|
license = licenses.asl20;
|
||
|
|
maintainers = with maintainers; [ lach ];
|
||
|
|
};
|
||
|
|
}
|