Skip to content

Commit a38a80a

Browse files
committed
Remove debug prints
1 parent d45222e commit a38a80a

File tree

1 file changed

+1
-6
lines changed

1 file changed

+1
-6
lines changed

src/tests/datasets/test_custom_dataset.py

Lines changed: 1 addition & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -132,12 +132,7 @@ def test_tokenize_dialog(tokenizer, monkeypatch, setup_tokenizer, llama_version)
132132
]
133133

134134
result = tokenize_dialog(dialog, tokenizer)
135-
print(f"{tokenizer.encode('system')=}")
136-
print(f"{tokenizer.encode('user')=}")
137-
print(f"{tokenizer.encode('assistant')=}")
138-
print(f"{tokenizer.decode(result['input_ids'])=}")
139-
print(f"{result['labels']=}")
140-
135+
141136
if "Llama-2" in llama_version:
142137
assert result["labels"][:12] == [-100] * 12
143138
assert result["labels"][17:28] == [-100] * 11

0 commit comments

Comments
 (0)