Remove old logic

Author: tomislav-peharda

labelbox/schema/model_run.py

@@ -26,6 +26,9 @@ def upsert_labels(self, label_ids, timeout_seconds=600):
             ID of newly generated async task
         """
 
+        if len(label_ids) < 1:
+            raise ValueError("Must provide at least one label id")
+
         sleep_time = 5
 
         mutation_name = 'createMEAModelRunLabelRegistrationTask'
@@ -49,21 +52,6 @@ def upsert_labels(self, label_ids, timeout_seconds=600):
 
             time.sleep(sleep_time)
 
-    def upsert_labels(self, label_ids):
-
-        if len(label_ids) < 1:
-            raise ValueError("Must provide at least one label id")
-
-        query_str = """mutation upsertModelRunLabelsPyApi($modelRunId: ID!, $labelIds : [ID!]!) {
-          upsertModelRunLabels(where : { id : $modelRunId}, data : {labelIds: $labelIds})}
-          """
-        res = self.client.execute(query_str, {
-            'modelRunId': self.uid,
-            'labelIds': label_ids
-        })
-        # TODO: Return a task
-        return True
-
     def add_predictions(
         self,
         name: str,