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