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):
7474 def __init__ (
7575 self ,
7676 train_tensors : BaseDatasetInputType ,
77- dataset_name : Optional [ str ] = None ,
77+ dataset_name : str = "" ,
7878 val_tensors : Optional [BaseDatasetInputType ] = None ,
7979 test_tensors : Optional [BaseDatasetInputType ] = None ,
8080 resampling_strategy : Union [CrossValTypes , HoldoutValTypes ] = HoldoutValTypes .holdout_validation ,
@@ -108,9 +108,9 @@ def __init__(
108108 val_transforms (Optional[torchvision.transforms.Compose]):
109109 Additional Transforms to be applied to the validation/test data
110110 """
111- self .dataset_name = ""
111+ self .dataset_name = dataset_name
112112
113- if self .dataset_name is not None :
113+ if self .dataset_name == "" :
114114 self .dataset_name = str (uuid .uuid1 (clock_seq = os .getpid ()))
115115
116116 if not hasattr (train_tensors [0 ], 'shape' ):
You can’t perform that action at this time.
0 commit comments