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

   1 {
   2   buildPythonPackage,
   3   fetchFromGitHub,
   4   fsspec,
   5   lib,
   6   numpy,
   7   pandas,
   8   pyarrow,
   9   pytestCheckHook,
  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   pythonRelaxDeps = [ "pyarrow" ];
  25
  26   propagatedBuildInputs = [
  27     fsspec
  28     numpy
  29     pandas
  30     pyarrow
  31   ];
  32
  33   nativeCheckInputs = [ pytestCheckHook ];
  34
  35   pythonImportsCheck = [ "embedding_reader" ];
  36
  37   meta = with lib; {
  38     description = "Efficiently read embedding in streaming from any filesystem";
  39     homepage = "https://github.com/rom1504/embedding-reader";
  40     license = licenses.mit;
  41     maintainers = with maintainers; [ samuela ];
  42   };
  43 }