commit 6eb25745769b364b79e5b474a2fd1ba0d72eb2b3 · pyrox.dev/nixpkgs

+5 -5
pkgs/development/python-modules/ctranslate2/default.nix
···

       16
       16
        
         pytestCheckHook,

     

       17
       17
        
         torch,

     

       18
       18
        
         transformers,

     

       19
       19
       +
         writableTmpDirAsHomeHook,

     

       19
       20
        
         wurlitzer,

     

       20
       21
        
       }:

     

       21
       22
        
       

     
···

       50
       51
        
           pytestCheckHook

     

       51
       52
        
           torch

     

       52
       53
        
           transformers

     

       54
       54
       +
           writableTmpDirAsHomeHook

     

       53
       55
        
           wurlitzer

     

       54
       56
        
         ];

     

       55
       57
        
       

     

       56
       58
        
         preCheck = ''

     

       57
       59
        
           # run tests against build result, not sources

     

       58
       60
        
           rm -rf ctranslate2

     

       59
       59
       -
       

     

       60
       60
       -
           export HOME=$TMPDIR

     

       61
       61
        
         '';

     

       62
       62
        
       

     

       63
       63
        
         disabledTests = lib.optionals stdenv.hostPlatform.isDarwin [

     
···

       72
       72
        
           "tests/test_transformers.py"

     

       73
       73
        
         ];

     

       74
       74
        
       

     

       75
       75
       -
         meta = with lib; {

     

       75
       75
       +
         meta = {

     

       76
       76
        
           description = "Fast inference engine for Transformer models";

     

       77
       77
        
           homepage = "https://github.com/OpenNMT/CTranslate2";

     

       78
       78
        
           changelog = "https://github.com/OpenNMT/CTranslate2/blob/${src.rev}/CHANGELOG.md";

     

       79
       79
       -
           license = licenses.mit;

     

       80
       80
       -
           maintainers = with maintainers; [ hexa ];

     

       79
       79
       +
           license = lib.licenses.mit;

     

       80
       80
       +
           maintainers = with lib.maintainers; [ hexa ];

     

       81
       81
        
         };

     

       82
       82
        
       }