typing

Author: sasha-scale

nucleus/model.py

@@ -28,7 +28,9 @@ def create_run(
         self,
         name: str,
         dataset: Dataset,
-        predictions: List[Union[BoxPrediction, PolygonPrediction]],
+        predictions: List[
+            Union[BoxPrediction, PolygonPrediction, SegmentationPrediction]
+        ],
         metadata: Optional[Dict] = None,
     ) -> ModelRun:
         payload: dict = {