build_dependencies:
  - pip
dependencies:
  - mlflow
  - cloudpickle==1.6.0
  - boto3
  - torchvision>=0.9.1
  - torch>=1.9.0