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