We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 729815b commit dec9614Copy full SHA for dec9614
keras_nlp/metrics/rouge_base.py
@@ -23,10 +23,9 @@
23
from keras_nlp.utils.tf_utils import tensor_to_string_list
24
25
try:
26
- import rouge_score
27
from rouge_score import rouge_scorer
28
except ImportError:
29
- rouge_score = None
+ rouge_scorer = None
30
31
32
class RougeBase(keras.metrics.Metric):
@@ -63,7 +62,7 @@ def __init__(
63
62
):
64
super().__init__(name=name, dtype=dtype, **kwargs)
65
66
- if rouge_score is None:
+ if rouge_scorer is None:
67
raise ImportError(
68
f"{self.__class__.__name__} requires the `rouge_score` "
69
"package. Please install it with `pip install rouge-score`."
0 commit comments