Traceback (most recent call last):
  File "flash_examples/finetuning/image_classification.py", line 14, in <module>
    import flash
  File "/home/eriba/software/lightning-flash/flash/__init__.py", line 50, in <module>
    from flash import tabular, text, vision
  File "/home/eriba/software/lightning-flash/flash/vision/__init__.py", line 1, in <module>
    from flash.vision.classification import ImageClassificationData, ImageClassifier
  File "/home/eriba/software/lightning-flash/flash/vision/classification/__init__.py", line 2, in <module>
    from flash.vision.classification.model import ImageClassifier
  File "/home/eriba/software/lightning-flash/flash/vision/classification/model.py", line 22, in <module>
    from flash.vision.backbones import backbone_and_num_features
  File "/home/eriba/software/lightning-flash/flash/vision/backbones.py", line 23, in <module>
    from pl_bolts.models.self_supervised import SimCLR, SwAV
  File "/home/eriba/software/venv-flash/lib/python3.8/site-packages/pl_bolts/__init__.py", line 47, in <module>
    from pl_bolts import callbacks, datamodules, datasets, losses, metrics, models, optimizers, transforms, utils
  File "/home/eriba/software/venv-flash/lib/python3.8/site-packages/pl_bolts/datamodules/__init__.py", line 2, in <module>
    from pl_bolts.datamodules.binary_mnist_datamodule import BinaryMNISTDataModule
  File "/home/eriba/software/venv-flash/lib/python3.8/site-packages/pl_bolts/datamodules/binary_mnist_datamodule.py", line 4, in <module>
    from pl_bolts.datasets import BinaryMNIST
  File "/home/eriba/software/venv-flash/lib/python3.8/site-packages/pl_bolts/datasets/__init__.py", line 11, in <module>
    from pl_bolts.datasets.imagenet_dataset import extract_archive, parse_devkit_archive, UnlabeledImagenet
  File "/home/eriba/software/venv-flash/lib/python3.8/site-packages/pl_bolts/datasets/imagenet_dataset.py", line 12, in <module>
    from torch._six import PY3
ImportError: cannot import name 'PY3' from 'torch._six' (/home/eriba/software/venv-flash/lib/python3.8/site-packages/torch/_six.py)