pkgs/development/python-modules/colbert-ai/default.nix

   1 {
   2   lib,
   3   buildPythonPackage,
   4   fetchPypi,
   5
   6   # build-system
   7   setuptools,
   8
   9   # dependencies
  10   bitarray,
  11   datasets,
  12   flask,
  13   python-dotenv,
  14   ninja,
  15   scipy,
  16   tqdm,
  17   transformers,
  18   ujson,
  19   gitpython,
  20   torch,
  21   faiss,
  22 }:
  23
  24 buildPythonPackage rec {
  25   pname = "colbert-ai";
  26   version = "0.2.21";
  27   pyproject = true;
  28
  29   src = fetchPypi {
  30     inherit version;
  31     pname = "colbert_ai";
  32     hash = "sha256-qNb9tOInLysI7Tf45QlgchYNhBXR5AWFdRiYt35iW6s=";
  33   };
  34
  35   pythonRemoveDeps = [ "git-python" ];
  36
  37   build-system = [
  38     setuptools
  39   ];
  40
  41   dependencies = [
  42     bitarray
  43     datasets
  44     faiss
  45     flask
  46     gitpython
  47     python-dotenv
  48     ninja
  49     scipy
  50     torch
  51     tqdm
  52     transformers
  53     ujson
  54   ];
  55
  56   pythonImportsCheck = [ "colbert" ];
  57
  58   # There is no tests
  59   doCheck = false;
  60
  61   meta = {
  62     description = "Fast and accurate retrieval model, enabling scalable BERT-based search over large text collections in tens of milliseconds";
  63     homepage = "https://github.com/stanford-futuredata/ColBERT";
  64     license = lib.licenses.mit;
  65     maintainers = with lib.maintainers; [
  66       bachp
  67     ];
  68   };
  69 }