yonikremer commited on
Commit
08b0161
1 Parent(s): af40811

disabled web search in tests

Browse files
Files changed (1) hide show
  1. tests.py +3 -3
tests.py CHANGED
@@ -37,15 +37,15 @@ def test_on_form_submit():
37
  model_name = "gpt2"
38
  output_length = 10
39
  prompt = "Answer yes or no, is the sky blue?"
40
- output = on_form_submit(model_name, output_length, prompt)
41
  assert output is not None
42
  assert len(output) > 0
43
  empty_prompt = ""
44
  with pytest.raises(ValueError):
45
- on_form_submit(model_name, output_length, empty_prompt)
46
  unsupported_model_name = "unsupported_model_name"
47
  with pytest.raises(ValueError):
48
- on_form_submit(unsupported_model_name, output_length, prompt)
49
 
50
 
51
  @pytest.mark.parametrize(
 
37
  model_name = "gpt2"
38
  output_length = 10
39
  prompt = "Answer yes or no, is the sky blue?"
40
+ output = on_form_submit(model_name, output_length, prompt, web_search=False)
41
  assert output is not None
42
  assert len(output) > 0
43
  empty_prompt = ""
44
  with pytest.raises(ValueError):
45
+ on_form_submit(model_name, output_length, empty_prompt, web_search=False)
46
  unsupported_model_name = "unsupported_model_name"
47
  with pytest.raises(ValueError):
48
+ on_form_submit(unsupported_model_name, output_length, prompt, web_search=False)
49
 
50
 
51
  @pytest.mark.parametrize(