Skip to content

GitLab

"examples/usage/git@developer.sourcefind.cn:change/sglang.git" did not exist on "a8552cb18b452e9a0a7e421651caf9d3c4deb673"
Commit 76da9765 authored by thomwolf's avatar thomwolf
Browse files

fix run_generation test

parent e691fc09
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
examples/test_examples.py
View file @ 76da9765
...@@ -103,7 +103,7 @@ class ExamplesTests(unittest.TestCase): ...@@ -103,7 +103,7 @@ class ExamplesTests(unittest.TestCase):
"--seed=42"] "--seed=42"]
model_type, model_name = ("--model_type=openai-gpt", model_type, model_name = ("--model_type=openai-gpt",
"--model_name_or_path=openai-gpt") "--model_name_or_path=openai-gpt")
with patch.object(sys, 'argv', testargs + [model_name]): with patch.object(sys, 'argv', testargs + [model_type, model_name]):
result = run_generation.main() result = run_generation.main()
self.assertGreaterEqual(len(result), 10) self.assertGreaterEqual(len(result), 10)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment