fixed issue with ray as a requirement
added:
- ray-default
- ray-tune
- torchmetrics
- timm
- pytorch
- torchvision
@@ -24,13 +24,18 @@ requirements:
- accelerate >=0.5.0
- lightning-flash >=0.5.1
- loguru >=0.5
- - ray >=1.6
+ - ray-default >=1.6.0
+ - ray-tune >=1.6.0
- rich
- smart_open >=5.1
+ - timm
+ - pytorch >=1.4
+ - torchvision
+ - torchmetrics >=0.6.0
test:
imports:
- - gradsflow
+ - {{ name|lower }}
commands:
- pip check
requires:
