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] }