enable windows test exec

* ensure temp file is closed for cache test * set torch_dtype to `float32` as default in atkgen * set torch_dtype to `float32` in huggingface tests Signed-off-by: Jeffrey Martin <[email protected]>
jmartin-tech · Jul 18, 2024 · 9de58af · 9de58af
1 parent aa2b851
commit 9de58af
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 4 deletions.
diff --git a/garak/probes/atkgen.py b/garak/probes/atkgen.py
@@ -57,7 +57,7 @@ class Tox(Probe):
         "red_team_model_type": "huggingface.Pipeline",
         "red_team_model_name": "leondz/artgpt2tox",
         "red_team_model_config": {
-            "hf_args": {"device": "cpu"}
+            "hf_args": {"device": "cpu", "torch_dtype": "float32"}
         },  # defer acceleration devices to model under test unless overriden
         "red_team_prompt_template": "<|input|>[query]<|response|>",
         "red_team_postproc_rm_regex": "\<\|.*",

diff --git a/tests/generators/test_huggingface.py b/tests/generators/test_huggingface.py
@@ -12,6 +12,7 @@ def hf_generator_config():
         "huggingface": {
             "hf_args": {
                 "device": "cpu",
+                "torch_dtype": "float32",
             },
         }
     }

diff --git a/tests/plugins/test_plugin_cache.py b/tests/plugins/test_plugin_cache.py
@@ -11,6 +11,7 @@ def temp_cache_location(request) -> None:
     with tempfile.NamedTemporaryFile(buffering=0, delete=False) as tmp:
         PluginCache._user_plugin_cache_file = tmp.name
         PluginCache._plugin_cache_file = tmp.name
+        tmp.close()
         os.remove(tmp.name)
     # reset the class level singleton
     PluginCache._plugin_cache_dict = None

diff --git a/tests/probes/test_probes_atkgen.py b/tests/probes/test_probes_atkgen.py
@@ -27,9 +27,7 @@ def test_atkgen_config():
         "generators": {
             rt_mod: {
                 rt_klass: {
-                    "hf_args": {
-                        "device": "cpu",
-                    },
+                    "hf_args": {"device": "cpu", "torch_dtype": "float32"},
                     "name": p.red_team_model_name,
                 }
             }