PeteBleackley commited on
Commit
be7beac
·
1 Parent(s): fcfc2b3

Forgot a len

Browse files
Files changed (1) hide show
  1. qarac/corpora/CorpusLoader.py +3 -3
qarac/corpora/CorpusLoader.py CHANGED
@@ -101,12 +101,12 @@ class CorpusLoader(object):
101
  yield (X,Y)
102
 
103
  def max_lengths(self):
104
- result = {column:max((row[column]
105
  for row in self.dataset))
106
  for column in self.text_inputs}
107
  for (column,(inside,outside)) in self.text_outputs.items():
108
- n = result[column] if column in result else max((len(row[column]
109
- for row in self.dataset)))
110
  result[inside] = n+1
111
  result[outside] = n+1
112
  return result
 
101
  yield (X,Y)
102
 
103
  def max_lengths(self):
104
+ result = {column:max((len(row[column])
105
  for row in self.dataset))
106
  for column in self.text_inputs}
107
  for (column,(inside,outside)) in self.text_outputs.items():
108
+ n = result[column] if column in result else max((len(row[column])
109
+ for row in self.dataset))
110
  result[inside] = n+1
111
  result[outside] = n+1
112
  return result