Spaces:
Sleeping
Sleeping
update formula matching
Browse files
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
|