update return types

Author: jihan-yin

nucleus/dataset.py

@@ -96,7 +96,7 @@ def autotag_items(self, autotag_name, for_scores_greater_than=0):
                 'autotagItems': {
                     ref_id: str,
                     score: float,
-                    model_prediction_id: str | None
+                    model_prediction_annotation_id: str | None
                     ground_truth_annotation_id: str | None,
                 }[],
                 'autotag': {
@@ -122,7 +122,7 @@ def autotag_training_items(self, autotag_name):
             {
                 'autotagPositiveTrainingItems': {
                     ref_id: str,
-                    model_prediction_id: str | None,
+                    model_prediction_annotation_id: str | None,
                     ground_truth_annotation_id: str | None,
                 }[],
                 'autotag': {