fix pickling issue

sfu-natlang · Sep 23, 2019 · 0ea0324 · 0ea0324
1 parent 901c5cb
commit 0ea0324
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/nd/dataset/vocab.py b/nd/dataset/vocab.py
@@ -61,7 +61,6 @@ def build_vocabs(path, lost_lang, known_lang, max_size=0):
 
 @dataclass(frozen=True, order=True)
 class Word:
-    __slots__ = ['lang', 'form', 'idx']
     lang: str
     form: str
     idx: int

diff --git a/nd/magic_tensor/core.py b/nd/magic_tensor/core.py
@@ -68,7 +68,12 @@ def __getattribute__(self, attr):
         try:
             return super().__getattribute__(attr)
         except AttributeError:
-            orig = getattr(self.tensor, attr)
+            try:
+                tensor = super().__getattribute__('_tensor')
+            except AttributeError:
+                raise
+
+            orig = getattr(tensor, attr)
             if attr in _SAFE_METHODS:
                 setattr(self, attr, orig)
             elif attr in _SAFE_METHODS_WITH_WRAPPER: