Spaces:

ChemFM
/

reaction_prediction

Sleeping

feiyang-cai commited on Jan 26

Commit

f0ad687

verified ·

1 Parent(s): fc058c4

Update utils.py

Files changed (1) hide show

utils.py CHANGED Viewed

@@ -61,7 +61,8 @@ class DataCollatorForCausalLMEval(object):
         print(instances)
         srcs = instances[0]['src']
-        task_type = instances[0]['task_type'][0]
         if task_type == 'retrosynthesis':
             src_start_str = self.product_start_str

         print(instances)
         srcs = instances[0]['src']
+        task_type = instances[0]['task_type']
+        print(task_type)
         if task_type == 'retrosynthesis':
             src_start_str = self.product_start_str