Skip to content

Commit aa0eec5

Browse files
authored
[feat] Add __str__ to autoPyTorchEnum (#405)
1 parent 048656e commit aa0eec5

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

autoPyTorch/utils/common.py

+3
Original file line numberDiff line numberDiff line change
@@ -101,6 +101,9 @@ def __eq__(self, other: Any) -> bool:
101101
def __hash__(self) -> int:
102102
return hash(self.value)
103103

104+
def __str__(self) -> str:
105+
return str(self.value)
106+
104107

105108
def custom_collate_fn(batch: List) -> List[Optional[torch.Tensor]]:
106109
"""

0 commit comments

Comments
 (0)