.item() added to the 'target' variable in fakedataset.py (#3587)

* added .item() for target inside getitem

* changed fakedataset's expected return type
Co-authored-by: Vasilis Vryniotis <datumbox@users.noreply.github.com>

test/test_datasets.py

@@ -1323,7 +1323,7 @@ class SBDatasetTestCase(datasets_utils.ImageDatasetTestCase):
 
 class FakeDataTestCase(datasets_utils.ImageDatasetTestCase):
     DATASET_CLASS = datasets.FakeData
-    FEATURE_TYPES = (PIL.Image.Image, torch.Tensor)
+    FEATURE_TYPES = (PIL.Image.Image, int)
 
     def dataset_args(self, tmpdir, config):
         return ()

torchvision/datasets/fakedata.py

@@ -60,7 +60,7 @@ class FakeData(VisionDataset):
         if self.target_transform is not None:
             target = self.target_transform(target)
 
-        return img, target
+        return img, target.item()
 
     def __len__(self) -> int:
         return self.size