summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--var/spack/repos/builtin/packages/py-timm/package.py4
1 files changed, 4 insertions, 0 deletions
diff --git a/var/spack/repos/builtin/packages/py-timm/package.py b/var/spack/repos/builtin/packages/py-timm/package.py
index c14681ea41..445eee2a5c 100644
--- a/var/spack/repos/builtin/packages/py-timm/package.py
+++ b/var/spack/repos/builtin/packages/py-timm/package.py
@@ -12,9 +12,13 @@ class PyTimm(PythonPackage):
homepage = "https://github.com/rwightman/pytorch-image-models"
pypi = "timm/timm-0.4.12.tar.gz"
+ version('0.5.4', sha256='5d7b92e66a76c432009aba90d515ea7a882aae573415a7c5269e3617df901c1f')
version('0.4.12', sha256='b14be70dbd4528b5ca8657cf5bc2672c7918c3d9ebfbffe80f4785b54e884b1e')
depends_on('python@3.6:', type=('build', 'run'))
depends_on('py-setuptools', type='build')
depends_on('py-torch@1.4:', type=('build', 'run'))
depends_on('py-torchvision', type=('build', 'run'))
+ # https://github.com/rwightman/pytorch-image-models/pull/1256
+ depends_on('pil@:9', type=('build', 'run'))
+ depends_on('py-numpy', type=('build', 'run'))