Browse Source

update DataSet使得可以与之前老版本的DataSet兼容

tags/v0.5.5
yh_cc 5 years ago
parent
commit
fbd2fd4ead
1 changed files with 12 additions and 1 deletions
  1. +12
    -1
      fastNLP/core/dataset.py

+ 12
- 1
fastNLP/core/dataset.py View File

@@ -406,7 +406,18 @@ class DataSet(object):


else: else:
raise ValueError("data only be dict or list type.") raise ValueError("data only be dict or list type.")
self.collater = Collater()
self._collater = Collater()

@property
def collater(self):
if self._collater is None:
self._collater = Collater()
return self._collater

@collater.setter
def collater(self, value):
assert isinstance(value, Collater)
self._collater = value


def __contains__(self, item): def __contains__(self, item):
return item in self.field_arrays return item in self.field_arrays


Loading…
Cancel
Save