1{
2 buildPythonPackage,
3 fetchFromGitHub,
4 fsspec,
5 lib,
6 numpy,
7 pandas,
8 pyarrow,
9 pytestCheckHook,
10}:
11
12buildPythonPackage 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}