Refactored conditional statement in langtest.py

JohnSnowLabs · Nov 9, 2023 · 6815260 · 6815260
1 parent 1f1932e
commit 6815260
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/langtest/langtest.py b/langtest/langtest.py
@@ -463,7 +463,7 @@ def report(
 
         summary = defaultdict(lambda: defaultdict(int))
 
-        if self.task == "political":
+        if self.task.category == "ideology":
             self.df_report = report.political_report(self._generated_results)
             return self.df_report