pytest
six>=1.13.0
wandb
comet_ml
