Optimize imports.

spring-media · Jun 21, 2021 · 1e034a8 · 1e034a8
1 parent 5580ad4
commit 1e034a8
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/dp/model/model.py b/dp/model/model.py
@@ -4,10 +4,10 @@
 
 import torch
 import torch.nn as nn
-from torch.nn import TransformerEncoderLayer, LayerNorm, TransformerEncoder, ModuleList
+from torch.nn import TransformerEncoderLayer, LayerNorm, TransformerEncoder
+
 from dp.model.utils import get_dedup_tokens, _make_len_mask, _generate_square_subsequent_mask, PositionalEncoding
 from dp.preprocessing.text import Preprocessor
-from dp.utils.io import read_config
 
 
 class ModelType(Enum):

diff --git a/run_prediction.py b/run_prediction.py
@@ -1,4 +1,3 @@
-import torch
 from dp.phonemizer import Phonemizer
 
 if __name__ == '__main__':