Import Errors in Datasets Class

[Ian-VA]

Hi,

When using the latest version of medicaltorch (or at least, the one installed by pip), importing the datasets class into the program raises the following error:

from torch._six import string_classes, int_classes                                   
ImportError: cannot import name 'int_classes' from 'torch._six'

I've found that this can be fixed by removing int_classes in the following line in datasets.py:

from torch._six import string_classes, int_classes

and, instead, declaring int_classes = int.

[tzuifurther]

In Pytorch 2.0 stable removes the _six, please see this and apex has some function to check the pytorch version this