Browse Source

fix dataset & instance

tags/v0.2.0
yunfan 7 years ago
parent
commit
5abd2bf4d5
2 changed files with 9 additions and 4 deletions
  1. +4
    -2
      fastNLP/core/dataset.py
  2. +5
    -2
      fastNLP/core/instance.py

+ 4
- 2
fastNLP/core/dataset.py View File

@@ -55,8 +55,10 @@ class DataSet(object):
raise AttributeError('{} does not exist.'.format(item))

def __setattr__(self, key, value):
self.__setitem__(key, value)

if hasattr(self, 'fields'):
self.__setitem__(key, value)
else:
super().__setattr__(self, key, value)

def __repr__(self):
return "\n".join(['{}: {}'.format(name, repr(self.dataset[name][self.idx])) for name


+ 5
- 2
fastNLP/core/instance.py View File

@@ -22,13 +22,16 @@ class Instance(object):
return self.add_field(name, field)

def __getattr__(self, item):
if item in self.fields:
if hasattr(self, 'fields') and item in self.fields:
return self.fields[item]
else:
raise AttributeError('{} does not exist.'.format(item))

def __setattr__(self, key, value):
self.__setitem__(key, value)
if hasattr(self, 'fields'):
self.__setitem__(key, value)
else:
super().__setattr__(key, value)

def __repr__(self):
return self.fields.__repr__()

Loading…
Cancel
Save