appease linter

NVIDIA · Jul 4, 2024 · 9bd86a4 · 9bd86a4
1 parent 24a4efb
commit 9bd86a4
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 6 deletions.
diff --git a/garak/probes/atkgen.py b/garak/probes/atkgen.py
@@ -65,7 +65,7 @@ class Tox(Probe):
     def probe(self, generator) -> List[garak.attempt.Attempt]:
         output_is_conversation = _config.system.verbose >= 2
 
-        if self.redteamer == None:
+        if self.redteamer is None:
 
             print(
                 "🔴🪖  ", end=""
@@ -103,10 +103,10 @@ def probe(self, generator) -> List[garak.attempt.Attempt]:
             turns = []
             logging.debug("atkgen: attempt %s uuid %s", i, this_attempt.uuid)
 
-            if output_is_conversation:
-                print("atkgen: 🆕 ⋅.˳˳.⋅ॱ˙˙ॱ New conversation ॱ˙˙ॱ⋅.˳˳.⋅ 🗣️")
-            else:
+            if not output_is_conversation:
                 t = tqdm(total=2 * self.max_calls, leave=False)
+            else:
+                print("atkgen: 🆕 ⋅.˳˳.⋅ॱ˙˙ॱ New conversation ॱ˙˙ॱ⋅.˳˳.⋅ 🗣️")
 
             while calls_made < self.max_calls and keep_going:
                 if not output_is_conversation:

diff --git a/tests/probes/test_probes_atkgen.py b/tests/probes/test_probes_atkgen.py
@@ -3,8 +3,6 @@
 
 import tempfile
 
-import pytest
-
 import garak._config
 import garak._plugins
 import garak.attempt