(cls)
| 47 | |
| 48 | @classmethod |
| 49 | def setUpClass(cls): |
| 50 | super().setUpClass() |
| 51 | cls.model = caching.CachingModelWrapper( |
| 52 | glue_models.SST2Model(BERT_TINY_PATH), 'test' |
| 53 | ) |
| 54 | cls.dataset = lit_dataset.IndexedDataset( |
| 55 | id_fn=caching.input_hash, |
| 56 | spec={ |
| 57 | 'sentence': lit_types.TextSegment(), |
| 58 | 'label': lit_types.CategoryLabel(vocab=['0', '1']) |
| 59 | }, |
| 60 | examples=_EXAMPLES, |
| 61 | ) |
| 62 | cls.model_outputs = list(cls.model.predict(_EXAMPLES)) |
| 63 | |
| 64 | def setUp(self): |
| 65 | super().setUp() |