File tree Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -74,7 +74,7 @@ class BaseDataset(Dataset, metaclass=ABCMeta):
74
74
def __init__ (
75
75
self ,
76
76
train_tensors : BaseDatasetInputType ,
77
- dataset_name : Optional [ str ] = None ,
77
+ dataset_name : str = "" ,
78
78
val_tensors : Optional [BaseDatasetInputType ] = None ,
79
79
test_tensors : Optional [BaseDatasetInputType ] = None ,
80
80
resampling_strategy : Union [CrossValTypes , HoldoutValTypes ] = HoldoutValTypes .holdout_validation ,
@@ -108,9 +108,9 @@ def __init__(
108
108
val_transforms (Optional[torchvision.transforms.Compose]):
109
109
Additional Transforms to be applied to the validation/test data
110
110
"""
111
- self .dataset_name = ""
111
+ self .dataset_name = dataset_name
112
112
113
- if self .dataset_name is not None :
113
+ if self .dataset_name == "" :
114
114
self .dataset_name = str (uuid .uuid1 (clock_seq = os .getpid ()))
115
115
116
116
if not hasattr (train_tensors [0 ], 'shape' ):
You can’t perform that action at this time.
0 commit comments