`Document.entities` field is unusable when using data from Classifier output

[evekhm]

Hello,

The wrapped_document, when using document.from_batch_process_metadata (or any other methods) will be missing entities field when using data from the Classifier.

When using output of splitter, everything works fine.
But with classifier - you wont get any important information like type and confidence.

from google.cloud.documentai_toolbox import document
import os

doc = document.Document.from_document_path(os.path.join(os.path.dirname(__file__), "output-document_split.json"))
print(doc.entities)
doc = document.Document.from_document_path(os.path.join(os.path.dirname(__file__), "output-document_classify.json"))
print(doc.entities)

output-document_split.json
output-document_classify.json

[evekhm]

I do see that information is there inside shards.entities, but entities itself is totally broken/missing/unusable

[evekhm]

Looking further at the issue:

Traceback (most recent call last):
  File "/Library/Frameworks/Python.framework/Versions/3.10/lib/python3.10/dataclasses.py", line 405, in wrapper
    result = user_function(self)
  File "<string>", line 3, in __repr__
AttributeError: 'Entity' object has no attribute 'start_page'

Both start_page and end_page need to be made Optional (since this info is not provided by the Classifier)