pkgs/development/python-modules/embedding-reader/default.nix

   1 { buildPythonPackage
   2 , fetchFromGitHub
   3 , fsspec
   4 , lib
   5 , numpy
   6 , pandas
   7 , pyarrow
   8 , pytestCheckHook
   9 , pythonRelaxDepsHook
  10 }:
  11
  12 buildPythonPackage rec {
  13   pname = "embedding-reader";
  14   version = "1.7.0";
  15   format = "setuptools";
  16
  17   src = fetchFromGitHub {
  18     owner = "rom1504";
  19     repo = pname;
  20     rev = "refs/tags/${version}";
  21     hash = "sha256-paN6rAyH3L7qCfWPr5kXo9Xl57gRMhdcDnoyLJ7II2w=";
  22   };
  23
  24   nativeBuildInputs = [ pythonRelaxDepsHook ];
  25
  26   pythonRelaxDeps = [ "pyarrow" ];
  27
  28   propagatedBuildInputs = [ fsspec numpy pandas pyarrow ];
  29
  30   nativeCheckInputs = [ pytestCheckHook ];
  31
  32   pythonImportsCheck = [ "embedding_reader" ];
  33
  34   meta = with lib; {
  35     description = "Efficiently read embedding in streaming from any filesystem";
  36     homepage = "https://github.com/rom1504/embedding-reader";
  37     license = licenses.mit;
  38     maintainers = with maintainers; [ samuela ];
  39   };
  40 }