Merge pull request #12 from pltrdy/revert-10-revert-8-hane1818-patch-1

Revert "Revert "Fix error to fit python2""
pltrdy · Mar 26, 2018 · 91529df · 91529df
2 parents 40b77b0 + 1a2218f
commit 91529df
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/rouge/rouge.py b/rouge/rouge.py
@@ -71,7 +71,7 @@ def __init__(self, metrics=None, stats=None):
                 raise ValueError("Unknown stat '%s'" % s)
 
     def get_scores(self, hyps, refs, avg=False):
-        if isinstance(hyps, string_types):
+        if isinstance(hyps, six.string_types):
             hyps, refs = [hyps], [refs]
 
         assert(type(hyps) == type(refs))