Code change

my_metric.py

@@ -13,8 +13,8 @@ Moses and PyTDC metrics
 
 _KWARGS_DESCRIPTION = """
 Args:
-    list_of_generated_smiles (`list` of `string`): Predicted labels.
-    list_of_test_smiles (`list` of `string`): test.
+    generated_smiles (`list` of `string`): Predicted labels.
+    train_smiles (`list` of `string`): test.
 
 Returns:
     All moses metrics
@@ -78,7 +78,7 @@ class my_metric(evaluate.Metric):
         Results = metrics.get_all_metrics(generated_smiles)
         
         evaluator = Evaluator(name = 'Diversity')
-        Diversity = evaluator(list_of_generated_smiles)
+        Diversity = evaluator(generated_smiles)
         
         evaluator = Evaluator(name = 'KL_Divergence')
         KL_Divergence = evaluator(generated_smiles, train_smiles)