summaryrefslogtreecommitdiff
path: root/tests/test-tokenizer-0.py
diff options
context:
space:
mode:
authorBrian <mofosyne@gmail.com>2024-05-05 15:07:48 +1000
committerGitHub <noreply@github.com>2024-05-05 08:07:48 +0300
commit6fbd43221167bf96112f899daf22c127b282cbcf (patch)
tree46a6c8d157adc202f7ac9fa46b85f5df50618c15 /tests/test-tokenizer-0.py
parent842500144ee02c8b0a46d2cc43880f8d80998fa5 (diff)
py : logging and flake8 suppression refactoring (#7081)
Set one as executable and add basicConfig() to another. Also added noqa tag to test scripts.
Diffstat (limited to 'tests/test-tokenizer-0.py')
-rw-r--r--tests/test-tokenizer-0.py10
1 files changed, 5 insertions, 5 deletions
diff --git a/tests/test-tokenizer-0.py b/tests/test-tokenizer-0.py
index 8e7638e4..cd760d1c 100644
--- a/tests/test-tokenizer-0.py
+++ b/tests/test-tokenizer-0.py
@@ -13,7 +13,7 @@ fname_tok = args.fname_tok
tokenizer = AutoTokenizer.from_pretrained(dir_tokenizer)
-print('tokenizing file: ', fname_tok)
+print('tokenizing file: ', fname_tok) # noqa: NP100
fname_out = fname_tok + '.tok'
with open(fname_tok, 'r', encoding='utf-8') as f:
lines = f.readlines()
@@ -21,7 +21,7 @@ with open(fname_tok, 'r', encoding='utf-8') as f:
t_start = time.time()
res = tokenizer.encode(s, add_special_tokens=False)
t_end = time.time()
- print('\nmain : tokenized in', "{:.3f}".format(1000.0 * (t_end - t_start)), 'ms (py)')
+ print('\nmain : tokenized in', "{:.3f}".format(1000.0 * (t_end - t_start)), 'ms (py)') # noqa: NP100
with open(fname_out, 'w', encoding='utf-8') as f:
for x in res:
# LLaMA v3 for some reason strips the space for these tokens (and others)
@@ -41,6 +41,6 @@ with open(fname_tok, 'r', encoding='utf-8') as f:
# f.write(str(x) + ' \'' + tokenizer.decode(x) + '\'\n')
# f.write(str(x) + ' \'' + tokenizer.decode(x).strip() + '\'\n')
f.write(str(x) + '\n')
- print('len(res): ', len(res))
- print('len(lines): ', len(lines))
-print('results written to: ', fname_out)
+ print('len(res): ', len(res)) # noqa: NP100
+ print('len(lines): ', len(lines)) # noqa: NP100
+print('results written to: ', fname_out) # noqa: NP100