chores: fixed small issue with start index problem (#56)

munkhuushmgl · web-flow · commit e1f2ffd94277 · 2020-11-11T14:15:02.000-08:00
* chores: fixed small issue with start index problem

* added missing update
diff --git a/documentai/snippets/batch_process_documents_sample_v1beta3.py b/documentai/snippets/batch_process_documents_sample_v1beta3.py
@@ -110,7 +110,7 @@ def get_text(doc_element: dict, document: dict):
     for segment in doc_element.text_anchor.text_segments:
         start_index = (
             int(segment.start_index)
-            if "start_index" in doc_element.text_anchor.__dict__
+            if segment in doc_element.text_anchor.text_segments
             else 0
         )
         end_index = int(segment.end_index)
diff --git a/documentai/snippets/process_document_sample_v1beta3.py b/documentai/snippets/process_document_sample_v1beta3.py
@@ -77,7 +77,7 @@ def get_text(doc_element: dict, document: dict):
     for segment in doc_element.text_anchor.text_segments:
         start_index = (
             int(segment.start_index)
-            if segment.start_index in doc_element.text_anchor.text_segments
+            if segment in doc_element.text_anchor.text_segments
             else 0
         )
         end_index = int(segment.end_index)
diff --git a/documentai/snippets/quickstart_sample_v1beta3.py b/documentai/snippets/quickstart_sample_v1beta3.py
@@ -70,7 +70,7 @@ def get_text(doc_element: dict, document: dict):
     for segment in doc_element.text_anchor.text_segments:
         start_index = (
             int(segment.start_index)
-            if segment.start_index in doc_element.text_anchor.text_segments
+            if segment in doc_element.text_anchor.text_segments
             else 0
         )
         end_index = int(segment.end_index)

Original file line number	Diff line number	Diff line change
`@@ -110,7 +110,7 @@ def get_text(doc_element: dict, document: dict):`
`110`	`110`	`for segment in doc_element.text_anchor.text_segments:`
`111`	`111`	`start_index = (`
`112`	`112`	`int(segment.start_index)`
`113`		`- if "start_index" in doc_element.text_anchor.__dict__`
	`113`	`+ if segment in doc_element.text_anchor.text_segments`
`114`	`114`	`else 0`
`115`	`115`	`)`
`116`	`116`	`end_index = int(segment.end_index)`
Original file line number	Diff line number	Diff line change
`@@ -77,7 +77,7 @@ def get_text(doc_element: dict, document: dict):`
`77`	`77`	`for segment in doc_element.text_anchor.text_segments:`
`78`	`78`	`start_index = (`
`79`	`79`	`int(segment.start_index)`
`80`		`- if segment.start_index in doc_element.text_anchor.text_segments`
	`80`	`+ if segment in doc_element.text_anchor.text_segments`
`81`	`81`	`else 0`
`82`	`82`	`)`
`83`	`83`	`end_index = int(segment.end_index)`
Original file line number	Diff line number	Diff line change
`@@ -70,7 +70,7 @@ def get_text(doc_element: dict, document: dict):`
`70`	`70`	`for segment in doc_element.text_anchor.text_segments:`
`71`	`71`	`start_index = (`
`72`	`72`	`int(segment.start_index)`
`73`		`- if segment.start_index in doc_element.text_anchor.text_segments`
	`73`	`+ if segment in doc_element.text_anchor.text_segments`
`74`	`74`	`else 0`
`75`	`75`	`)`
`76`	`76`	`end_index = int(segment.end_index)`