Test SFTData initialization
(self)
| 104 | cls.temp_dir.cleanup() |
| 105 | |
| 106 | def test_SFTData_initialization(self): |
| 107 | """Test SFTData initialization""" |
| 108 | dataset = SFTData( |
| 109 | train_file=self.csv_file, |
| 110 | tokenizer=self.tokenizer, |
| 111 | max_len=128, |
| 112 | sample=1, |
| 113 | seed=0, |
| 114 | category="games" |
| 115 | ) |
| 116 | self.assertEqual(len(dataset), 1) |
| 117 | self.assertTrue(hasattr(dataset, 'inputs')) |
| 118 | |
| 119 | def test_D3Dataset_initialization(self): |
| 120 | """Test D3Dataset initialization""" |