lfoppiano commited on
Commit
9026ed3
1 Parent(s): cf77f40

update formula matching

Browse files
Files changed (1) hide show
  1. document_qa/grobid_processors.py +7 -1
document_qa/grobid_processors.py CHANGED
@@ -467,6 +467,11 @@ class GrobidMaterialsProcessor(BaseProcessor):
467
  if status != 200:
468
  result = []
469
 
 
 
 
 
 
470
  compositions = []
471
  for material in result:
472
  if 'resolvedFormulas' in material:
@@ -476,7 +481,8 @@ class GrobidMaterialsProcessor(BaseProcessor):
476
  elif 'formula' in material:
477
  if 'formulaComposition' in material['formula']:
478
  compositions.append(material['formula']['formulaComposition'])
479
-
 
480
  return compositions
481
 
482
  @staticmethod
 
467
  if status != 200:
468
  result = []
469
 
470
+ compositions = self.output_info(result)
471
+
472
+ return compositions
473
+
474
+ def output_info(self, result):
475
  compositions = []
476
  for material in result:
477
  if 'resolvedFormulas' in material:
 
481
  elif 'formula' in material:
482
  if 'formulaComposition' in material['formula']:
483
  compositions.append(material['formula']['formulaComposition'])
484
+ if 'name' in material:
485
+ compositions.append(material['name'])
486
  return compositions
487
 
488
  @staticmethod