fix tests

Author: Matt Sokoloff

labelbox/schema/model.py

@@ -8,15 +8,12 @@ class Model(DbObject):
         can make predictions on new data.
         Attributes:
             name (str)
-            ontology (Relationship): `ToOne` relationship to Ontology
-            model_runs (Relationship): `ToMany` relationship to ModelRun (TODO)
-            slices (Relationship): `ToMany` relationship to Slice (TODO)
+            model_runs (Relationship): `ToMany` relationship to ModelRun
         """
 
     name = Field.String("name")
-
     model_runs = Relationship.ToMany("ModelRun", False)
-    ontology = Relationship.ToOne("Ontology", False)
+
 
     def create_model_run(self, name):
         """ Creates a model run belonging to this model.

tests/integration/conftest.py

@@ -6,6 +6,7 @@
 from random import randint
 from string import ascii_letters
 from types import SimpleNamespace
+import time
 import uuid
 import os
 import re
@@ -333,6 +334,7 @@ def configured_project_with_label(client, rand_gen, annotation_submit_fn):
     upload_task = MALPredictionImport.create_from_objects(
         client, project.uid, f'mal-import-{uuid.uuid4()}', predictions)
     upload_task.wait_until_done()
+    time.sleep(1)
     annotation_submit_fn(project.uid, data_row.uid)
     yield project
     dataset.delete()

tests/integration/test_model.py

@@ -11,12 +11,10 @@ def test_model(client, configured_project, rand_gen):
     data = {"name": rand_gen(str), "ontology_id": ontology.uid}
     model = client.create_model(data["name"], data["ontology_id"])
     assert model.name == data["name"]
-    assert model.ontology().uid == data["ontology_id"]
 
     after = list(client.get_models())
     assert len(after) == len(before) + 1
     assert model in after
 
     model = client.get_model(model.uid)
     assert model.name == data["name"]
-    assert model.ontology().uid == data["ontology_id"]

tests/integration/test_model_run.py

@@ -11,9 +11,8 @@ def test_model_run(client, configured_project_with_label, rand_gen):
     model_run = model.create_model_run(name)
     assert model_run.name == name
     assert model_run.model_id == model.uid
-    # TODO: Uncomment this out once https://github.com/Labelbox/labelbox-python/pull/189 is merged.
-    #assert model_run.created_by_id == client.get_user(
-    #).uid
+    assert model_run.created_by_id == client.get_user(
+    ).uid
 
     label = list(project.labels())[0]
     model_run.upsert_labels([label.uid])