Skip to content

Commit 5283c3a

Browse files
committed
update ggml temp file
Signed-off-by: Jeffrey Martin <jemartin@nvidia.com>
1 parent 2a906d8 commit 5283c3a

1 file changed

Lines changed: 8 additions & 4 deletions

File tree

tests/generators/test_ggml.py

Lines changed: 8 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -37,17 +37,21 @@ def test_init_missing_model():
3737

3838

3939
def test_init_bad_model():
40-
with tempfile.NamedTemporaryFile(mode="w", suffix="_test_model.gguf") as file:
40+
with tempfile.NamedTemporaryFile(
41+
mode="w", suffix="_test_model.gguf", encoding="utf-8", delete=False
42+
) as file:
4143
file.write(file.name)
42-
file.seek(0)
44+
file.close()
4345
with pytest.raises(RuntimeError) as exc_info:
4446
garak.generators.ggml.GgmlGenerator(file.name)
47+
os.remove(file.name)
4548
assert "not in GGUF" in str(exc_info.value)
4649

4750

4851
def test_init_good_model():
49-
with tempfile.NamedTemporaryFile(suffix="_test_model.gguf") as file:
52+
with tempfile.NamedTemporaryFile(suffix="_test_model.gguf", delete=False) as file:
5053
file.write(garak.generators.ggml.GGUF_MAGIC)
51-
file.seek(0)
54+
file.close()
5255
g = garak.generators.ggml.GgmlGenerator(file.name)
56+
os.remove(file.name)
5357
assert type(g) is garak.generators.ggml.GgmlGenerator

0 commit comments

Comments
 (0)