fix bug

Author: Matt Sokoloff

labelbox/data/annotation_types/classification/classification.py

@@ -56,7 +56,7 @@ class Checklist(_TempName):
 
     """
     name: Literal["checklist"] = "checklist"
-    answer: ClassificationAnswer
+    answer: List[ClassificationAnswer]
 
 
 class Text(BaseModel):

labelbox/data/serialization/labelbox_v1/classification.py

@@ -9,13 +9,13 @@
 
 
 class LBV1ClassificationAnswer(LBV1Feature):
-    def to_common(self, answer: "LBV1ClassificationAnswer") -> ClassificationAnswer:
-        return ClassificationAnswer(feature_schema_id=answer.schema_id,
-                  name=answer.title,
-                  keyframe = answer.keyframe,
+    def to_common(self) -> ClassificationAnswer:
+        return ClassificationAnswer(feature_schema_id=self.schema_id,
+                  name=self.title,
+                  keyframe = self.keyframe,
                   extra={
-                      'feature_id': answer.feature_id,
-                      'value': answer.value
+                      'feature_id': self.feature_id,
+                      'value': self.value
                   })
 
     @classmethod

tests/data/annotation_types/classification/test_classification.py

@@ -125,7 +125,6 @@ def test_checklist():
 
     with pytest.raises(ValidationError):
         classification = Checklist(answer=answer)
-    breakpoint()
 
     classification = Checklist(answer=[answer])
     assert classification.dict() == {