public/nixpkgs
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

torchvision: update reason why tests fails

Browse Source
This commit is contained in:
Jörg Thalheim 2020-12-30 07:42:16 +01:00
parent 77c70abb81
commit d198cbabe7
No known key found for this signature in database
GPG Key ID: 003F2096411B5F92
1 changed files with 8 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
pkgs/development/python-modules/torchvision/default.nix
View File

@ -9,7 +9,7 @@
, scipy , scipy
, pillow , pillow
, pytorch , pytorch
, pytestCheckHook , pytest
}: }:
buildPythonPackage rec { buildPythonPackage rec {
@ -25,19 +25,21 @@ buildPythonPackage rec {
nativeBuildInputs = [ libpng ninja which ]; nativeBuildInputs = [ libpng ninja which ];
TORCHVISION_INCLUDE="${libjpeg_turbo.dev}/include/"; TORCHVISION_INCLUDE = "${libjpeg_turbo.dev}/include/";
TORCHVISION_LIBRARY="${libjpeg_turbo}/lib/"; TORCHVISION_LIBRARY = "${libjpeg_turbo}/lib/";
buildInputs = [ libjpeg_turbo libpng ]; buildInputs = [ libjpeg_turbo libpng ];
propagatedBuildInputs = [ numpy pillow pytorch scipy ]; propagatedBuildInputs = [ numpy pillow pytorch scipy ];
# checks fail to import _C.so and various other failures # tries to download many datasets for tests
doCheck = false; doCheck = false;
checkInputs = [ pytestCheckHook ]; checkPhase = ''
HOME=$TMPDIR py.test test --ignore=test/test_datasets_download.py
'';
pytestFlagsArray = [ "--ignore=test/test_datasets_download.py" ]; checkInputs = [ pytest ];
meta = with stdenv.lib; { meta = with stdenv.lib; {
description = "PyTorch vision library"; description = "PyTorch vision library";