fix issue with truncated dict

28069cf4 · alexeib · Myle Ott · cfd2a3a0 · 28069cf4
Commit 28069cf4 authored Sep 21, 2018 by alexeib Committed by Myle Ott Sep 25, 2018
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 3 deletions

fairseq/data/dictionary.py fairseq/data/dictionary.py +3 -3

No files found.
--- a/fairseq/data/dictionary.py
+++ b/fairseq/data/dictionary.py
@@ -203,9 +203,9 @@ class Dictionary(object):
 class TruncatedDictionary(object):

    def __init__(self, wrapped_dict, length):
-        self.__class__ = type(dict.__class__.__name__,
-                              (self.__class__, dict.__class__), {})
-        self.__dict__ = dict.__dict__
+        self.__class__ = type(wrapped_dict.__class__.__name__,
+                              (self.__class__, wrapped_dict.__class__), {})
+        self.__dict__ = wrapped_dict.__dict__
        self.wrapped_dict = wrapped_dict
        self.length = min(len(self.wrapped_dict), length)