From c3e394d0caefaa6f104b79e251d702d0b73f3ce1 Mon Sep 17 00:00:00 2001 From: marcsello Date: Fri, 29 May 2020 17:50:18 +0200 Subject: [PATCH] Fixed boolean memes --- classification_service/mule.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/classification_service/mule.py b/classification_service/mule.py index 4655595..f637429 100644 --- 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] }