saicharan2804 commited on
Commit
e426e35
1 Parent(s): 73b91cc

Code change

Browse files
Files changed (1) hide show
  1. my_metric.py +4 -6
my_metric.py CHANGED
@@ -56,18 +56,16 @@ _CITATION = """
56
 
57
 
58
  @evaluate.utils.file_utils.add_start_docstrings(_DESCRIPTION, _KWARGS_DESCRIPTION)
59
- class my_metric(evaluate.Metric):
60
  def _info(self):
61
  return evaluate.MetricInfo(
62
  description=_DESCRIPTION,
63
  citation=_CITATION,
64
  inputs_description=_KWARGS_DESCRIPTION,
65
- features=datasets.Features(
66
-
67
  "generated_smiles": datasets.Value("string"),
68
- "train_smiles": datasets.Value("string"),
69
-
70
- ),
71
 
72
  reference_urls=["https://github.com/molecularsets/moses", "https://tdcommons.ai/functions/oracles/"],
73
  )
 
56
 
57
 
58
  @evaluate.utils.file_utils.add_start_docstrings(_DESCRIPTION, _KWARGS_DESCRIPTION)
59
+ class my_metric(evaluate.Measurement):
60
  def _info(self):
61
  return evaluate.MetricInfo(
62
  description=_DESCRIPTION,
63
  citation=_CITATION,
64
  inputs_description=_KWARGS_DESCRIPTION,
65
+ features=datasets.Features({
 
66
  "generated_smiles": datasets.Value("string"),
67
+ "train_smiles": datasets.Value("string"),
68
+ }),
 
69
 
70
  reference_urls=["https://github.com/molecularsets/moses", "https://tdcommons.ai/functions/oracles/"],
71
  )