Merge pull request #75 from scaleapi/jihan/validation-fix

Nucleus - prediction pdf optional

Author: jihan-yin

nucleus/prediction.py

@@ -75,6 +75,7 @@ def to_payload(self) -> dict:
         payload = super().to_payload()
         if self.confidence is not None:
             payload[CONFIDENCE_KEY] = self.confidence
+        if self.class_pdf is not None:
             payload[CLASS_PDF_KEY] = self.class_pdf
 
         return payload
@@ -93,7 +94,7 @@ def from_json(cls, payload: dict):
             confidence=payload.get(CONFIDENCE_KEY, None),
             annotation_id=payload.get(ANNOTATION_ID_KEY, None),
             metadata=payload.get(METADATA_KEY, {}),
-            class_pdf=payload.get(CLASS_PDF_KEY, {}),
+            class_pdf=payload.get(CLASS_PDF_KEY, None),
         )
 
 
@@ -124,6 +125,7 @@ def to_payload(self) -> dict:
         payload = super().to_payload()
         if self.confidence is not None:
             payload[CONFIDENCE_KEY] = self.confidence
+        if self.class_pdf is not None:
             payload[CLASS_PDF_KEY] = self.class_pdf
 
         return payload
@@ -139,5 +141,5 @@ def from_json(cls, payload: dict):
             confidence=payload.get(CONFIDENCE_KEY, None),
             annotation_id=payload.get(ANNOTATION_ID_KEY, None),
             metadata=payload.get(METADATA_KEY, {}),
-            class_pdf=payload.get(CLASS_PDF_KEY, {}),
+            class_pdf=payload.get(CLASS_PDF_KEY, None),
         )

pyproject.toml

@@ -21,7 +21,7 @@ exclude = '''
 
 [tool.poetry]
 name = "scale-nucleus"
-version = "0.1.8"
+version = "0.1.9"
 description = "The official Python client library for Nucleus, the Data Platform for AI"
 license =  "MIT"
 authors = ["Scale AI Nucleus Team <nucleusapi@scaleapi.com>"]

tests/helpers.py

@@ -103,6 +103,11 @@ def reference_id_from_url(url):
         "confidence": 0.10 * i,
         "class_pdf": TEST_BOX_MODEL_PDF,
     }
+    if i != 0
+    else {
+        **TEST_BOX_ANNOTATIONS[i],
+        "confidence": 0.10 * i,
+    }
     for i in range(len(TEST_BOX_ANNOTATIONS))
 ]
 
@@ -112,6 +117,11 @@ def reference_id_from_url(url):
         "confidence": 0.10 * i,
         "class_pdf": TEST_POLYGON_MODEL_PDF,
     }
+    if i != 0
+    else {
+        **TEST_POLYGON_ANNOTATIONS[i],
+        "confidence": 0.10 * i,
+    }
     for i in range(len(TEST_POLYGON_ANNOTATIONS))
 ]