Revised API to align with recent changes

This commit is contained in:
kinaryml 2022-10-25 23:13:12 -07:00
parent 0de97497fa
commit 75af46d273
2 changed files with 3 additions and 9 deletions

View File

@ -85,8 +85,6 @@ class ClassificationList:
"""
classifications: List[Classification]
tensor_index: Optional[int] = None
tensor_name: Optional[str] = None
@doc_controls.do_not_generate_docs
def to_pb2(self) -> _ClassificationListProto:
@ -95,9 +93,7 @@ class ClassificationList:
classification=[
classification.to_pb2()
for classification in self.classifications
],
tensor_index=self.tensor_index,
tensor_name=self.tensor_name)
])
@classmethod
@doc_controls.do_not_generate_docs
@ -110,9 +106,7 @@ class ClassificationList:
classifications=[
Classification.create_from_pb2(classification)
for classification in pb2_obj.classification
],
tensor_index=pb2_obj.tensor_index,
tensor_name=pb2_obj.tensor_name)
])
def __eq__(self, other: Any) -> bool:
"""Checks if this object is equal to the given object.

View File

@ -80,7 +80,7 @@ def _get_expected_gesture_recognition_result(
classifications=[
_Classification(label=gesture_label, index=gesture_index,
display_name='')
], tensor_index=0, tensor_name='')
])
return _GestureRecognitionResult(
gestures=[gesture],
handedness=[landmarks_detection_result.classifications],