added suggestions from review
@@ -23,7 +23,6 @@ requirements:
run:
- python >=3.7
- accelerate >=0.5.0
- # - lightning-flash >=0.5.1
- loguru >=0.5
- ray-default >=1.6.0
- ray-tune >=1.6.0
@@ -33,19 +32,16 @@ requirements:
- pytorch >=1.4
- torchvision
- torchmetrics >=0.6.0
- run_constrained:
- - comet_ml
test:
imports:
- - {{ name|lower }}
+ - gradsflow
commands:
- pip check
- - pip install comet-ml
requires:
- pip
- - lightning-flash
- - pytorch-lightning
+ # - lightning-flash
+ # - pytorch-lightning
about:
home: https://github.com/gradsflow/gradsflow
