datasets.concatenate_datasets() does not seem to work for this since features do not match. Also BertForNextSentencePrediction expects a file_path. Initially I thought it was a wrapper which can take datasets objects.