commit f5f1911538ac1f0bc4bfadda9335e1b6a365db2c · pyrox.dev/nixpkgs

+32 -9
pkgs/development/python-modules/clarifai/default.nix
···

       1
       1
        
       {

     

       2
       2
        
         lib,

     

       3
       3
       +
         aiohttp,

     

       3
       4
        
         buildPythonPackage,

     

       4
       5
        
         clarifai-grpc,

     

       5
       6
        
         clarifai-protocol,

     

       6
       7
        
         click,

     

       7
       8
        
         fetchFromGitHub,

     

       8
       9
        
         fsspec,

     

       10
       10
       +
         huggingface-hub,

     

       9
       11
        
         inquirerpy,

     

       10
       12
        
         numpy,

     

       11
       13
        
         pillow,

     

       14
       14
       +
         pkgs,

     

       15
       15
       +
         psutil,

     

       12
       16
        
         pycocotools,

     

       17
       17
       +
         pydantic-core,

     

       18
       18
       +
         pytest-asyncio,

     

       13
       19
        
         pytestCheckHook,

     

       14
       20
        
         pythonOlder,

     

       15
       21
        
         pyyaml,

     

       16
       22
        
         rich,

     

       23
       23
       +
         ruff,

     

       17
       24
        
         schema,

     

       18
       25
        
         setuptools,

     

       19
       26
        
         tabulate,

     

       20
       27
        
         tqdm,

     

       21
       28
        
         tritonclient,

     

       29
       29
       +
         uv,

     

       30
       30
       +
         writableTmpDirAsHomeHook,

     

       22
       31
        
       }:

     

       23
       32
        
       

     

       24
       33
        
       buildPythonPackage rec {

     

       25
       34
        
         pname = "clarifai";

     

       26
       26
       -
         version = "11.6.7";

     

       35
       35
       +
         version = "11.8.2";

     

       27
       36
        
         pyproject = true;

     

       28
       37
        
       

     

       29
       38
        
         disabled = pythonOlder "3.8";

     
···

       32
       41
        
           owner = "Clarifai";

     

       33
       42
        
           repo = "clarifai-python";

     

       34
       43
        
           tag = version;

     

       35
       35
       -
           hash = "sha256-1ftwsIKJ494F8q45x0LtvOZhM72AAhJWe0LligNNpkQ=";

     

       44
       44
       +
           hash = "sha256-7luvK7xnkJRALXm7X7mg6VG9qRpq9pmFdBRW2GP3heI=";

     

       36
       45
        
         };

     

       37
       46
        
       

     

       38
       47
        
         pythonRelaxDeps = [

     

       48
       48
       +
           "clarifai-protocol"

     

       39
       49
        
           "click"

     

       40
       50
        
           "fsspec"

     

       51
       51
       +
           "ruff"

     

       41
       52
        
           "schema"

     

       53
       53
       +
           "uv"

     

       42
       54
        
         ];

     

       43
       55
        
       

     

       44
       56
        
         build-system = [ setuptools ];

     

       45
       57
        
       

     

       46
       58
        
         dependencies = [

     

       59
       59
       +
           aiohttp

     

       47
       60
        
           clarifai-grpc

     

       48
       61
        
           clarifai-protocol

     

       49
       62
        
           click

     
···

       51
       64
        
           inquirerpy

     

       52
       65
        
           numpy

     

       53
       66
        
           pillow

     

       67
       67
       +
           psutil

     

       68
       68
       +
           pydantic-core

     

       54
       69
        
           pyyaml

     

       55
       70
        
           rich

     

       71
       71
       +
           ruff

     

       56
       72
        
           schema

     

       57
       73
        
           tabulate

     

       58
       74
        
           tqdm

     

       59
       75
        
           tritonclient

     

       76
       76
       +
           uv

     

       60
       77
        
         ];

     

       61
       78
        
       

     

       62
       79
        
         optional-dependencies = {

     

       63
       80
        
           all = [ pycocotools ];

     

       64
       81
        
         };

     

       65
       82
        
       

     

       66
       66
       -
         nativeCheckInputs = [ pytestCheckHook ];

     

       67
       67
       -
       

     

       68
       68
       -
         preCheck = ''

     

       69
       69
       -
           export HOME=$(mktemp -d)

     

       70
       70
       -
         '';

     

       83
       83
       +
         nativeCheckInputs = [

     

       84
       84
       +
           pkgs.gitMinimal

     

       85
       85
       +
           huggingface-hub

     

       86
       86
       +
           pytest-asyncio

     

       87
       87
       +
           pytestCheckHook

     

       88
       88
       +
           writableTmpDirAsHomeHook

     

       89
       89
       +
         ];

     

       71
       90
        
       

     

       72
       91
        
         disabledTests = [

     

       73
       92
        
           # Test requires network access and API key

     
···

       79
       98
        
         disabledTestPaths = [

     

       80
       99
        
           # Tests require network access and API key

     

       81
       100
        
           "tests/cli/test_compute_orchestration.py"

     

       82
       82
       -
           "tests/runners/test_anymodel.py"

     

       83
       101
        
           "tests/runners/test_download_checkpoints.py"

     

       102
       102
       +
           "tests/runners/test_model_run_locally.py"

     

       103
       103
       +
           "tests/runners/test_model_upload.py"

     

       104
       104
       +
           "tests/runners/test_num_threads_config.py"

     

       105
       105
       +
           "tests/runners/test_runners_proto.py"

     

       84
       106
        
           "tests/runners/test_runners.py"

     

       85
       85
       -
           "tests/runners/test_textmodel.py"

     

       86
       107
        
           "tests/runners/test_url_fetcher.py"

     

       108
       108
       +
           "tests/runners/test_vllm_model_upload.py"

     

       87
       109
        
           "tests/test_app.py"

     

       88
       110
        
           "tests/test_data_upload.py"

     

       89
       111
        
           "tests/test_eval.py"

     

       112
       112
       +
           "tests/test_list_models.py"

     

       90
       113
        
           "tests/test_model_predict.py"

     

       91
       114
        
           "tests/test_model_train.py"

     

       92
       115
        
           "tests/test_rag.py"