Update namespaces inside torch.utils.data to the latest. (#13167)

* Update torch.utils.data namespaces to the latest. * Format * Update Dataloader. * Style
2021-08-19 20:29:51 +08:00
parent 1fec32adc6
commit 91ff480e26
24 changed files with 41 additions and 44 deletions
--- a/tests/test_trainer_tpu.py
+++ b/tests/test_trainer_tpu.py
@@ -32,7 +32,7 @@ logger = logging.get_logger(__name__)
 if is_torch_available():
    import torch
    from torch import nn
-    from torch.utils.data.dataset import Dataset
+    from torch.utils.data import Dataset

    from transformers import Trainer