Skip to content

Commit

Permalink
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
[fix] Address the francisco's review
Browse files Browse the repository at this point in the history
nabenabe0928 committed Apr 14, 2021

Verified

This commit was created on GitHub.com and signed with GitHub’s verified signature.
1 parent d20c1b5 commit 244a23e
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions autoPyTorch/datasets/base_dataset.py
Original file line number Diff line number Diff line change
@@ -74,7 +74,7 @@ class BaseDataset(Dataset, metaclass=ABCMeta):
def __init__(
self,
train_tensors: BaseDatasetInputType,
dataset_name: Optional[str] = None,
dataset_name: str = "",
val_tensors: Optional[BaseDatasetInputType] = None,
test_tensors: Optional[BaseDatasetInputType] = None,
resampling_strategy: Union[CrossValTypes, HoldoutValTypes] = HoldoutValTypes.holdout_validation,
@@ -108,9 +108,9 @@ def __init__(
val_transforms (Optional[torchvision.transforms.Compose]):
Additional Transforms to be applied to the validation/test data
"""
self.dataset_name = ""
self.dataset_name = dataset_name

if self.dataset_name is not None:
if self.dataset_name == "":
self.dataset_name = str(uuid.uuid1(clock_seq=os.getpid()))

if not hasattr(train_tensors[0], 'shape'):

0 comments on commit 244a23e

Please sign in to comment.