Method apply

lit_gpt/datamodules/base_test.py:16–17

(self, prompt, **kwargs)

def test_sft_dataset(max_seq_length, ignore_index, mask_prompt, MOCK_TOKENIZER_FIXTURE):
    class Style(PromptStyle):
        def apply(self, prompt, **kwargs):
            return f'In: {prompt} Out:'

    i = ignore_index
    data = [