summaryrefslogtreecommitdiff
path: root/Omni/Bild/Deps/outlines.nix
blob: 6426c15d6050dda72390b669e721868f197fdbfa (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
{
  lib,
  sources,
  buildPythonPackage,
  interegular,
  jinja2,
  lark,
  numpy,
  perscache,
  pillow,
  pydantic,
  regex,
  scipy,
  tenacity,
  torch,
}:
buildPythonPackage rec {
  pname = "outlines";
  version = sources.outlines.rev;
  format = "pyproject";

  src = sources.outlines;

  propagatedBuildInputs = [
    interegular
    jinja2
    lark
    numpy
    perscache
    pillow
    pydantic
    regex
    scipy
    tenacity
    torch
  ];

  doCheck = false; # no tests currently
  pythonImportsCheck = ["outlines"];

  meta = with lib; {
    description = "Probabilistic Generative Model Programming";
    homepage = "https://github.com/normal-computing/outlines";
    license = licenses.asl20;
    maintainers = with maintainers; [bsima];
  };
}