diff --git a/autoPyTorch/utils/common.py b/autoPyTorch/utils/common.py index 4e4d4e804..a443068f5 100644 --- a/autoPyTorch/utils/common.py +++ b/autoPyTorch/utils/common.py @@ -101,6 +101,9 @@ def __eq__(self, other: Any) -> bool: def __hash__(self) -> int: return hash(self.value) + def __str__(self) -> str: + return str(self.value) + def custom_collate_fn(batch: List) -> List[Optional[torch.Tensor]]: """