Sort imports for optional third-party libraries.

These libraries aren't always installed in the virtual environment where
isort is running. Declaring them properly avoids mixing these
third-party imports with local imports.
This commit is contained in:
Aymeric Augustin
2019-12-22 11:17:48 +01:00
parent 2a34d5b71b
commit c11b3e2926
10 changed files with 28 additions and 16 deletions

View File

@@ -19,6 +19,7 @@ import math
import os
import time
import psutil
import torch
import torch.nn as nn
import torch.nn.functional as F
@@ -27,7 +28,6 @@ from torch.utils.data import BatchSampler, DataLoader, RandomSampler
from torch.utils.data.distributed import DistributedSampler
from tqdm import tqdm
import psutil
from grouped_batch_sampler import GroupedBatchSampler, create_lengths_groups
from lm_seqs_dataset import LmSeqsDataset
from transformers import get_linear_schedule_with_warmup