def METHOD_NAME(self, *args, **kwargs):
    ans = self.dataset.METHOD_NAME(*args, **kwargs)
    if "net_input" in ans:
        ans["net_input"]["target_language_id"] = self.target_language_id
        ans["net_input"]["dataset_name"] = self.name
    return ans