Method change_dataset

pythia/tasks/base_task.py:163–167

(self)

        return self._preprocess_item(item)

    def change_dataset(self):
        self.dataset_choice = np.random.choice(
            self.num_datasets, 1, p=self.dataset_probablities
        )[0]
        self.chosen_dataset = self.datasets[self.dataset_choice]

    def verbose_dump(self, *args, **kwargs):
        self.chosen_dataset.verbose_dump(*args, **kwargs)