diff --git a/pkgs/development/python-modules/clarifai/default.nix b/pkgs/development/python-modules/clarifai/default.nix index 8f6fdc18509ff..1d541e3dab724 100644 --- a/pkgs/development/python-modules/clarifai/default.nix +++ b/pkgs/development/python-modules/clarifai/default.nix @@ -2,38 +2,75 @@ , buildPythonPackage , fetchFromGitHub , pythonOlder +, setuptools , clarifai-grpc +, numpy +, opencv4 +, pillow +, pyyaml +, rich +, schema +, tqdm +, tritonclient , pytestCheckHook }: buildPythonPackage rec { pname = "clarifai"; - version = "9.7.1"; - format = "setuptools"; + version = "9.10.4"; + pyproject = true; disabled = pythonOlder "3.8"; src = fetchFromGitHub { owner = "Clarifai"; - repo = "clarifai-python-utils"; + repo = "clarifai-python"; rev = "refs/tags/${version}"; - hash = "sha256-/zgHgD2kf3ZG7Mu9AEBfOwqpcD0Ye0LVrFxLeuGurCM="; + hash = "sha256-i/B4wtSqG/ZysAcidsXnqYPdcmu93Ufd6N9fxUmDJ2E="; }; + nativeBuildInputs = [ + setuptools + ]; + propagatedBuildInputs = [ clarifai-grpc + numpy + tqdm + opencv4 + tritonclient + rich + schema + pillow + pyyaml ]; nativeCheckInputs = [ pytestCheckHook ]; + disabledTests = [ + # require network access and API key + "test_export_workflow_general" + ]; + + disabledTestPaths = [ + # require network access and API key + "tests/test_app.py" + "tests/test_data_upload.py" + "tests/test_model_predict.py" + "tests/test_model_train.py" + "tests/test_search.py" + "tests/workflow/test_create_delete.py" + "tests/workflow/test_predict.py" + ]; + pythonImportsCheck = [ "clarifai" ]; meta = with lib; { description = "Clarifai Python Utilities"; - homepage = "https://github.com/Clarifai/clarifai-python-utils"; - changelog = "https://github.com/Clarifai/clarifai-python-utils/releases/tag/${src.rev}"; + homepage = "https://github.com/Clarifai/clarifai-python"; + changelog = "https://github.com/Clarifai/clarifai-python/releases/tag/${src.rev}"; license = licenses.asl20; maintainers = with maintainers; [ natsukium ]; };