Spaces:

saicharan2804
/

molgenevalmetric

Sleeping

saicharan2804 commited on Apr 3, 2024

Commit

e8da7e0

1 Parent(s): 2225f77

Fixed error

Files changed (1) hide show

molgenevalmetric.py CHANGED Viewed

@@ -402,13 +402,13 @@ class molgenevalmetric(evaluate.Metric):
             inputs_description=_KWARGS_DESCRIPTION,
             features=datasets.Features(
                 {
-                    "gen": datasets.Sequence(datasets.Value("string")),
-                    "train": datasets.Sequence(datasets.Value("string")),
                 }
                 if self.config_name == "multilabel"
                 else {
-                    "gen": datasets.Value("string"),
-                    "train": datasets.Value("string"),
                 }
             ),

             inputs_description=_KWARGS_DESCRIPTION,
             features=datasets.Features(
                 {
+                    "gensmi": datasets.Sequence(datasets.Value("string")),
+                    "trainsmi": datasets.Sequence(datasets.Value("string")),
                 }
                 if self.config_name == "multilabel"
                 else {
+                    "gensmi": datasets.Value("string"),
+                    "trainsmi": datasets.Value("string"),
                 }
             ),