From 029bb41b55d4cb756e01d3b176f6bc96add05c39 Mon Sep 17 00:00:00 2001 From: Amir MOHAMMADI <amir.mohammadi@idiap.ch> Date: Mon, 26 Aug 2019 12:25:34 +0200 Subject: [PATCH] [bob-devel] More tests for pytorch --- conda/beat-devel/meta.yaml | 4 ++-- conda/bob-devel/meta.yaml | 6 +++--- conda/bob-devel/run_test.py | 19 +++++++++++++++++++ 3 files changed, 24 insertions(+), 5 deletions(-) create mode 100644 conda/bob-devel/run_test.py diff --git a/conda/beat-devel/meta.yaml b/conda/beat-devel/meta.yaml index dfd3ea3c..ad9b6914 100644 --- a/conda/beat-devel/meta.yaml +++ b/conda/beat-devel/meta.yaml @@ -2,7 +2,7 @@ {% set nopin = ['ca-certificates', 'certifi', 'openssl', 'libstdcxx-ng', 'libgcc-ng', 'libcxx', 'libcxxabi'] %} package: name: beat-devel - version: 2019.08.14 + version: 2019.08.26 build: # the build number must be zero all the time. @@ -15,7 +15,7 @@ build: requirements: host: - python {{ python }} - - bob-devel 2019.08.14 + - bob-devel 2019.08.26 - docker-py 4.0.2 - flask 1.0.2 - flask-cors 3.0.7 diff --git a/conda/bob-devel/meta.yaml b/conda/bob-devel/meta.yaml index c64c1bb2..0eec1f79 100644 --- a/conda/bob-devel/meta.yaml +++ b/conda/bob-devel/meta.yaml @@ -4,7 +4,7 @@ package: name: bob-devel # please update the version of beat-devel and update bob-devel's version # there as well in the same merge request. - version: 2019.08.14 + version: 2019.08.26 build: # the build number must be zero all the time. @@ -94,11 +94,11 @@ requirements: {% endfor %} test: - imports: - - torch # [linux] requires: + - numpy - ffmpeg - pytorch # [linux] + - torchvision # [linux] commands: # we expect these features from ffmpeg: - ffmpeg -codecs | grep "DEVI.S zlib" # [unix] diff --git a/conda/bob-devel/run_test.py b/conda/bob-devel/run_test.py new file mode 100644 index 00000000..a09d245e --- /dev/null +++ b/conda/bob-devel/run_test.py @@ -0,0 +1,19 @@ +import sys + +# couple of imports to see if packages are working +import numpy + + +def test_pytorch(): + import torch + from torchvision.models import DenseNet + + model = DenseNet() + t = torch.randn(1, 3, 224, 224) + out = model(t) + assert out.shape[1] == 1000 + + +if sys.platform.startswith("linux"): + # test if pytorch installation is sane + test_pytorch() -- GitLab