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

* chores: fixed small issue with start index problem * added missing update
GoogleCloudPlatform · Nov 11, 2020 · 71987a6 · 71987a6
1 parent d1e4a53
commit 71987a6
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/document_ai/snippets/batch_process_documents_sample_v1beta3.py b/document_ai/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/document_ai/snippets/process_document_sample_v1beta3.py b/document_ai/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/document_ai/snippets/quickstart_sample_v1beta3.py b/document_ai/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)