From 68cea928ce913d4eaf0ce87914b505e04bbed59f Mon Sep 17 00:00:00 2001 From: Zeeshan Khan Suri <5270999+zshn25@users.noreply.github.com> Date: Mon, 20 Nov 2023 12:26:55 +0100 Subject: [PATCH 1/2] made root vaiable optional for Vision Dataset --- torchvision/datasets/vision.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/torchvision/datasets/vision.py b/torchvision/datasets/vision.py index aba19369b64..064b5114a23 100644 --- a/torchvision/datasets/vision.py +++ b/torchvision/datasets/vision.py @@ -12,7 +12,7 @@ class VisionDataset(data.Dataset): It is necessary to override the ``__getitem__`` and ``__len__`` method. Args: - root (string): Root directory of dataset. + root (string, optional): Root directory of dataset. transforms (callable, optional): A function/transforms that takes in an image and a label and returns the transformed versions of both. transform (callable, optional): A function/transform that takes in an PIL image @@ -29,7 +29,7 @@ class VisionDataset(data.Dataset): def __init__( self, - root: str, + root: Optional[str] = None, transforms: Optional[Callable] = None, transform: Optional[Callable] = None, target_transform: Optional[Callable] = None, From 703f8937aa14ae37f3c77cfd63279b82c8f040b6 Mon Sep 17 00:00:00 2001 From: Nicolas Hug Date: Mon, 20 Nov 2023 12:00:41 +0000 Subject: [PATCH 2/2] Update torchvision/datasets/vision.py --- torchvision/datasets/vision.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/torchvision/datasets/vision.py b/torchvision/datasets/vision.py index 064b5114a23..a672be38bf9 100644 --- a/torchvision/datasets/vision.py +++ b/torchvision/datasets/vision.py @@ -12,7 +12,7 @@ class VisionDataset(data.Dataset): It is necessary to override the ``__getitem__`` and ``__len__`` method. Args: - root (string, optional): Root directory of dataset. + root (string, optional): Root directory of dataset. Only used for `__repr__`. transforms (callable, optional): A function/transforms that takes in an image and a label and returns the transformed versions of both. transform (callable, optional): A function/transform that takes in an PIL image