Skip to content

Commit c8299e2

Browse files
committed
[fix] Fix an error due to the change in extract learning curve
1 parent fd1d179 commit c8299e2

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

autoPyTorch/evaluation/utils.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -129,10 +129,10 @@ def extract_learning_curve(stack: List[RunValue], key: Optional[str] = None) ->
129129
key = 'loss' if key is None else key
130130

131131
for entry in stack:
132-
info = entry.additional_info
133132
try:
133+
info = entry.additional_info
134134
learning_curve.append(getattr(entry, key, info[key]))
135-
except TypeError: # additional info is not dict
135+
except AttributeError: # additional info is not RunValue
136136
pass
137137
except KeyError: # Key does not exist
138138
pass

0 commit comments

Comments
 (0)