diff --git a/s1.py b/s1.py index cb3d90e..7ed9062 100755 --- a/s1.py +++ b/s1.py @@ -36,7 +36,7 @@ def get_rank_thing(filename, **kwargs): if result: # print(u'%s > %s' % (r[0], list(result))) count += 1 - owriter.writerow([','.join(list(result)), cnt] + w) + owriter.writerow([mn, ','.join(list(result)), cnt] + w) if cnt not in model_count[mn]: model_count[mn][cnt] = set() model_count[mn][cnt] = model_count[mn][cnt].union(result)