Skip to content

Commit 65b207e

Browse files
Update emotion assertion in test_inference.py
1 parent 3638798 commit 65b207e

1 file changed

Lines changed: 1 addition & 2 deletions

File tree

tests/test_inference.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -84,8 +84,7 @@ def predict(self, x, verbose=0):
8484

8585
result = predict_emotion(img)
8686

87-
assert result == "Happy"
88-
87+
assert result["emotion"] == "Happy"
8988
# =========================
9089
# Prediction output validity
9190
# =========================

0 commit comments

Comments
 (0)