Skip to content

Commit 70f8c7e

Browse files
Update emotion prediction assertion in tests
1 parent c650c1d commit 70f8c7e

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

tests/test_inference.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -120,7 +120,7 @@ def predict(self, x, verbose=0):
120120
r1 = predict_emotion(img)
121121
r2 = predict_emotion(img)
122122

123-
assert r1 == r2
123+
assert r1["emotion"] == r2["emotion"]
124124
assert r1 == "Neutral"
125125

126126
# =========================

0 commit comments

Comments
 (0)