Fix explanation

hupe1980 · Apr 20, 2024 · 26b17fd · 26b17fd
1 parent ec4c8e4
commit 26b17fd
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/aisploit/classifiers/package_hallucination.py b/aisploit/classifiers/package_hallucination.py
@@ -40,7 +40,11 @@ def score(self, input: str) -> Score[List[str]]:
             flagged=len(hallucinated_package) > 0,
             value=hallucinated_package,
             description="Return True if hallucinated packages are found in the input",
-            explanation="Did not find token in input",
+            explanation=(
+                f"Found {len(hallucinated_package)} packages in input"
+                if len(hallucinated_package) > 0
+                else "Did not find hallucinated packages in input"
+            ),
         )
 
     def _get_imported_packages(self, input: str) -> List[str]: