Remove rouge package check
It is now a project dependency.
@@ -65,13 +65,6 @@ class ExternalEvaluator(object):
class ROUGEEvaluator(ExternalEvaluator):
"""ROUGE evaluator based on https://github.com/pltrdy/rouge."""
- def __init__(self, labels_file=None, output_dir=None):
- try:
- import rouge # pylint: disable=unused-variable
- except ImportError:
- raise ImportError("Please install the 'rouge' package, run 'pip install rouge==0.3.0'")
- super(ROUGEEvaluator, self).__init__(labels_file=labels_file, output_dir=output_dir)
-
def name(self):
return "ROUGE"
