yonikremer commited on
Commit
59660ba
1 Parent(s): b426c55

deleting an assert that should be false

Browse files
Files changed (1) hide show
  1. tests.py +0 -1
tests.py CHANGED
@@ -21,7 +21,6 @@ def test_create_pipeline():
21
  pipeline: GroupedSamplingPipeLine = create_pipeline("gpt2")
22
  assert pipeline is not None
23
  assert pipeline.model_name == "gpt2"
24
- assert pipeline.wrapped_model.group_size == 5
25
  assert pipeline.wrapped_model.end_of_sentence_stop is False
26
  del pipeline
27
 
 
21
  pipeline: GroupedSamplingPipeLine = create_pipeline("gpt2")
22
  assert pipeline is not None
23
  assert pipeline.model_name == "gpt2"
 
24
  assert pipeline.wrapped_model.end_of_sentence_stop is False
25
  del pipeline
26