From f65d939f9a47781874ed4e42ef8257a124369a9c Mon Sep 17 00:00:00 2001
From: Giovanni Campagna <gcampagn@cs.stanford.edu>
Date: Wed, 13 Mar 2019 16:34:12 -0700
Subject: [PATCH] fix

---
 decanlp/utils/embeddings.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/decanlp/utils/embeddings.py b/decanlp/utils/embeddings.py
index 808ba646..ec6e9123 100644
--- a/decanlp/utils/embeddings.py
+++ b/decanlp/utils/embeddings.py
@@ -41,6 +41,8 @@ ENTITIES = ['DATE', 'DURATION', 'EMAIL_ADDRESS', 'HASHTAG',
 MAX_ARG_VALUES = 5
 
 class AlmondEmbeddings(torchtext.vocab.Vectors):
+    def __init__(self, name=None, cache=None, **kw):
+        super().__init__(name, cache, **kw)
 
     def cache(self, name, cache, url=None):
         del name
@@ -60,7 +62,7 @@ class AlmondEmbeddings(torchtext.vocab.Vectors):
 
         self.itos = itos
         self.stoi = {word: i for i, word in enumerate(itos)}
-        self.vectors = torch.Tensor(vectors).view(-1, dim)
+        self.vectors = torch.stack(vectors, dim=0).view(-1, dim)
         self.dim = dim