add some basic tests
{% set name = "vit-pytorch" %}
-{% set version = "0.25.1" %}
+{% set version = "0.25.3" %}
package:
name: "{{ name|lower }}"
@@ -7,7 +7,7 @@ package:
source:
url: "https://pypi.io/packages/source/{{ name[0] }}/{{ name }}/{{ name }}-{{ version }}.tar.gz"
- sha256: 70afcc360deb91a99cce18978440bf51d164c21afca777f7c70b09162d392f91
+ sha256: 2c9c71ddafbb390f0385ecd3ed728f3bc78678b5e89420c25f6360f6d4034b52
build:
number: 0
@@ -24,6 +24,16 @@ requirements:
- torch >=1.6
- torchvision
+test:
+ imports:
+ - vit_pytorch
+ source_files:
+ - tests/test.py
+ requires:
+ - pytest
+ commands:
+ - pytest tests/
+
about:
home: "https://github.com/lucidrains/vit-pytorch"
license: MIT
