Update test.py

test.py

@@ -58,12 +58,12 @@ BAD_WORDS_URL = "http://url/to/external/resource/bad_words.txt"
 
 
 @evaluate.utils.file_utils.add_start_docstrings(_DESCRIPTION, _KWARGS_DESCRIPTION)
-class test(evaluate.Metric):
+class test(evaluate.EvaluationModule):
     """TODO: Short description of my metric."""
 
     def _info(self):
         # TODO: Specifies the evaluate.MetricInfo object
-        return evaluate.MetricInfo(
+        return evaluate.EvaluationModule(
             # This is the description that will appear on the metrics page.
             description=_DESCRIPTION,
             citation=_CITATION,