Skip to content

Commit e7bd80a

Browse files
author
sfluegel
committed
fix: ignore empty list in evaluation
1 parent 7639179 commit e7bd80a

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

chebai/result/utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -155,7 +155,7 @@ def evaluate_model(
155155
test_labels = _concat_tuple(labels_list)
156156
return test_preds, test_labels
157157
return test_preds, None
158-
elif preds_list i
158+
elif len(preds_list) < 0:
159159
torch.save(
160160
_concat_tuple(preds_list),
161161
os.path.join(buffer_dir, f"preds{save_ind:03d}.pt"),

0 commit comments

Comments
 (0)