Fixed boolean memes

2020-05-29 17:50:18 +02:00 · 2020-05-29 17:50:18 +02:00 · c3e394d0ca
parent 88ddc35741
commit c3e394d0ca
1 changed files with 1 additions and 1 deletions
--- a/classification_service/mule.py
+++ b/classification_service/mule.py
@ -70,7 +70,7 @@ def run_classification(task, target_class_name: str):
    results = {
        "tag": task['tag'],
        "model": task['model'],
-        "is_target": class_id == target_id,
+        "is_target": bool(class_id == target_id),
        "probability": probability[target_id]
    }