python311Packages.fastembed: fix build

2024-02-14 15:11:48 +01:00
parent 06e6a7f4e5
commit 7832456e4c
1 changed files with 14 additions and 2 deletions
@@ -1,8 +1,11 @@
 { lib
 , buildPythonPackage
 , fetchFromGitHub
+, huggingface-hub
 , pythonOlder
+, pythonRelaxDepsHook
 , poetry-core
+, onnx
 , onnxruntime
 , requests
 , tokenizers
@@ -13,7 +16,7 @@
 buildPythonPackage rec {
  pname = "fastembed";
  version = "0.1.2";
-  format = "pyproject";
+  pyproject = true;

  disabled = pythonOlder "3.8";

@@ -26,16 +29,25 @@ buildPythonPackage rec {

  nativeBuildInputs = [
    poetry-core
+    pythonRelaxDepsHook
  ];

  propagatedBuildInputs = [
+    huggingface-hub
+    onnx
    onnxruntime
    requests
    tokenizers
    tqdm
  ];

-  pythonImportsCheck = [ "fastembed" ];
+  pythonImportsCheck = [
+    "fastembed"
+  ];
+
+  pythonRelaxDeps = [
+    "huggingface-hub"
+  ];

  nativeCheckInputs = [
    pytestCheckHook