pkgs/development/python-modules/outlines/default.nix

   1 { lib
   2 , buildPythonPackage
   3 , fetchFromGitHub
   4 , setuptools
   5 , setuptools-scm
   6 , interegular
   7 , cloudpickle
   8 , diskcache
   9 , joblib
  10 , jsonschema
  11 , pydantic
  12 , lark
  13 , nest-asyncio
  14 , numba
  15 , scipy
  16 , torch
  17 , transformers
  18 }:
  19
  20 buildPythonPackage rec {
  21   pname = "outlines";
  22   version = "0.0.38";
  23   pyproject = true;
  24
  25   src = fetchFromGitHub {
  26     owner = "outlines-dev";
  27     repo = pname;
  28     rev = "refs/tags/${version}";
  29     hash = "sha256-kOd2J6PV2idWaltkGUVNXfB+1dNU1f8vbnHQ5i1WxG8=";
  30   };
  31
  32   nativeBuildInputs = [
  33     setuptools
  34     setuptools-scm
  35   ];
  36
  37   propagatedBuildInputs = [
  38     interegular
  39     cloudpickle
  40     diskcache
  41     joblib
  42     jsonschema
  43     pydantic
  44     lark
  45     nest-asyncio
  46     numba
  47     scipy
  48     torch
  49     transformers
  50   ];
  51
  52   pythonImportsCheck = [
  53     "outlines"
  54   ];
  55
  56   meta = with lib; {
  57     description = "Structured text generation";
  58     homepage = "https://github.com/outlines-dev/outlines";
  59     license = licenses.asl20;
  60     maintainers = with maintainers; [ lach ];
  61   };
  62 }