stuff

mlmc/representation/representations.py

@@ -12,6 +12,7 @@ import tempfile
 import torch
 from io import BytesIO
 from transformers import *
+import pathlib
 
 
 EMBEDDINGCACHE = Path.home() / ".mlmc" / "embedding"
@@ -228,7 +229,6 @@ def get_embedding(name, **kwargs):
     tokenizer = get_white_space_tokenizer(vocabulary)
     return e, tokenizer
 
-import pathlib
 
 def get_transformer(model="bert", **kwargs):
     """

requirements.txt

@@ -18,5 +18,4 @@ datasketch
 torch>=1.5.0
 beautifulsoup4~=4.8.2
 requests~=2.23.0
-dgl~=0.5.2
-apex~=0.1
\ No newline at end of file
+dgl~=0.5.2
\ No newline at end of file