meta.yaml 1.61 KB
Newer Older
Guillaume HEUSCH's avatar
Guillaume HEUSCH committed
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25
{% set name = 'bob.learn.pytorch' %}
{% set project_dir = environ.get('RECIPE_DIR') + '/..' %}

package:
  name: {{ name }}
  version: {{ environ.get('BOB_PACKAGE_VERSION', '0.0.1') }}

build:
  skip: true  # [not linux]
  entry_points:
    - train_cnn.py = bob.learn.pytorch.scripts.train_cnn:main
  number: {{ environ.get('BOB_BUILD_NUMBER', 0) }}
  run_exports:
    - {{ pin_subpackage(name) }}
  script:
    - cd {{ project_dir }}
    {% if environ.get('BUILD_EGG') %}
    - python setup.py sdist --formats=zip
    {% endif %}
    - python setup.py install --single-version-externally-managed --record record.txt

requirements:
  host:
    - python {{ python }}
    - setuptools {{ setuptools }}
Amir MOHAMMADI's avatar
Amir MOHAMMADI committed
26 27
    - numpy >=1.11
    - docopt
Guillaume HEUSCH's avatar
Guillaume HEUSCH committed
28 29 30 31
    - bob.extension
    - bob.core
    - bob.io.base
    - bob.io.image
32
    - bob.bio.base
33
    - pytorch >=0.4
34
    - torchvision >=0.2.0
Guillaume HEUSCH's avatar
Guillaume HEUSCH committed
35 36 37
  run:
    - python
    - setuptools
38
    - numpy >=1.11
Amir MOHAMMADI's avatar
Amir MOHAMMADI committed
39
    - docopt
Guillaume HEUSCH's avatar
Guillaume HEUSCH committed
40
    - pytorch =0.4
41
    - torchvision >=0.2.0
Amir MOHAMMADI's avatar
Amir MOHAMMADI committed
42
    - matplotlib
Olegs NIKISINS's avatar
Olegs NIKISINS committed
43
    - h5py
44

Guillaume HEUSCH's avatar
Guillaume HEUSCH committed
45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64
test:
  imports:
    - {{ name }}
  commands:
    - train_cnn.py --help
    - nosetests --with-coverage --cover-package={{ name }} -sv {{ name }}
    - sphinx-build -aEW {{ project_dir }}/doc {{ project_dir }}/sphinx
    - sphinx-build -aEb doctest {{ project_dir }}/doc sphinx
    - conda inspect linkages -p $PREFIX {{ name }}  # [not win]
    - conda inspect objects -p $PREFIX {{ name }}  # [osx]
  requires:
    - bob-devel {{ bob_devel }}.*
    - nose
    - coverage
    - sphinx
    - sphinx_rtd_theme

about:
  home: https://www.idiap.ch/software/bob/
  license: BSD License
Amir MOHAMMADI's avatar
Amir MOHAMMADI committed
65 66
  summary: Bob bindings for PyTorch
  license_family: BSD