update requirements and remove unneeded imports

Author: nreimers

requirements.txt

@@ -1,6 +1,6 @@
-transformers>=3.1.0,<3.5.0
+transformers>=3.1.0,<3.6.0
 tqdm
-torch>=1.2.0
+torch>=1.6.0
 numpy
 scikit-learn
 scipy

sentence_transformers/evaluation/BinaryClassificationEvaluator.py

@@ -1,9 +1,5 @@
-from . import SentenceEvaluator, SimilarityFunction
-import torch
-from torch.utils.data import DataLoader
+from . import SentenceEvaluator
 import logging
-from tqdm import tqdm
-from sentence_transformers.util import batch_to_device
 import os
 import csv
 from sklearn.metrics.pairwise import paired_cosine_distances, paired_euclidean_distances, paired_manhattan_distances

sentence_transformers/evaluation/EmbeddingSimilarityEvaluator.py

@@ -1,5 +1,4 @@
 from . import SentenceEvaluator, SimilarityFunction
-import torch
 import logging
 import os
 import csv

sentence_transformers/evaluation/TripletEvaluator.py

@@ -1,9 +1,5 @@
 from . import SentenceEvaluator, SimilarityFunction
-import torch
-from torch.utils.data import DataLoader
 import logging
-from tqdm import tqdm
-from ..util import batch_to_device
 import os
 import csv
 from sklearn.metrics.pairwise import paired_cosine_distances, paired_euclidean_distances, paired_manhattan_distances

sentence_transformers/models/WeightedLayerPooling.py

@@ -4,11 +4,7 @@
 from typing import Union, Tuple, List, Iterable, Dict
 import os
 import json
-import numpy as np
-import torch.nn.functional as F
 
-from sklearn.metrics.pairwise import cosine_similarity
-from sklearn.preprocessing import normalize
 
 class WeightedLayerPooling(nn.Module):
     """

setup.py

@@ -18,9 +18,9 @@
     download_url="https://github.com/UKPLab/sentence-transformers/archive/v0.3.8.zip",
     packages=find_packages(),
     install_requires=[
-        'transformers>=3.1.0,<3.5.0',
+        'transformers>=3.1.0,<3.6.0',
         'tqdm',
-        'torch>=1.2.0',
+        'torch>=1.6.0',
         'numpy',
         'scikit-learn',
         'scipy',