Fix another test

Ubuntu · Ubuntu · commit 1acdf126ad4c · 2021-09-21T19:49:14.000Z
diff --git a/tests/test_prediction.py b/tests/test_prediction.py
@@ -23,6 +23,7 @@
     DatasetItem,
     ModelRun,
     Segment,
+    Point,
 )
 from nucleus.constants import ERROR_PAYLOAD
 

Original file line number	Diff line number	Diff line change
`@@ -23,6 +23,7 @@`
`23`	`23`	`DatasetItem,`
`24`	`24`	`ModelRun,`
`25`	`25`	`Segment,`
	`26`	`+ Point,`
`26`	`27`	`)`
`27`	`28`	`from nucleus.constants import ERROR_PAYLOAD`
`28`	`29`