def METHOD_NAME(self):
    model = BertModel(BertConfig())
    # construct sequence data
    input_ids, attention_mask = torch.tensor(
        [[3, 16, 87, 3, 9], [72, 12, 4, 90, 54]], dtype=torch.long
    ), torch.tensor([[1, 1, 1, 1, 1], [1, 1, 1, 1, 1]], dtype=torch.long)
    dataset = torch.utils.data.TensorDataset(input_ids, attention_mask)
    self.model_context = ModelContext(
        model=model, optim_func=torch.optim.Adam, optim_args={"lr": 1e-3}, dataset=dataset
    )
    self.analyser = get_analyser()