From 7039a35e7b9ebbf1f14f4669259560d7100c184c Mon Sep 17 00:00:00 2001
From: Mike <45373284+munkhuushmgl@users.noreply.github.com>
Date: Wed, 11 Nov 2020 14:15:02 -0800
Subject: [PATCH] chores: fixed small issue with start index problem (#56)

* chores: fixed small issue with start index problem

* added missing update
---
 samples/snippets/batch_process_documents_sample_v1beta3.py | 2 +-
 samples/snippets/process_document_sample_v1beta3.py        | 2 +-
 samples/snippets/quickstart_sample_v1beta3.py              | 2 +-
 3 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/samples/snippets/batch_process_documents_sample_v1beta3.py b/samples/snippets/batch_process_documents_sample_v1beta3.py
index 2936b3b3..6e22e0ea 100644
--- a/samples/snippets/batch_process_documents_sample_v1beta3.py
+++ b/samples/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/samples/snippets/process_document_sample_v1beta3.py b/samples/snippets/process_document_sample_v1beta3.py
index 330e8183..e666affa 100644
--- a/samples/snippets/process_document_sample_v1beta3.py
+++ b/samples/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/samples/snippets/quickstart_sample_v1beta3.py b/samples/snippets/quickstart_sample_v1beta3.py
index c5cd34ae..7eb4a8ff 100644
--- a/samples/snippets/quickstart_sample_v1beta3.py
+++ b/samples/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)