Merge pull request #212 from Labelbox/ms/remove-predictions

remove predictions and prediction models

Author: msokoloff1

docs/source/index.rst

@@ -102,22 +102,6 @@ Organization
    :members:
    :show-inheritance:
 
-Prediction
----------------------------------
-
-.. automodule:: labelbox.schema.prediction
-   :members: Prediction
-   :exclude-members: PredictionModel
-   :show-inheritance:
-
-PredictionModel
----------------------------------
-.. automodule:: labelbox.schema.prediction
-    :members: PredictionModel
-    :exclude-members: Prediction
-    :show-inheritance:
-    :noindex:
-
 Project
 ------------------------------
 

labelbox/__init__.py

@@ -15,7 +15,6 @@
 from labelbox.schema.asset_metadata import AssetMetadata
 from labelbox.schema.asset_attachment import AssetAttachment
 from labelbox.schema.webhook import Webhook
-from labelbox.schema.prediction import Prediction, PredictionModel
 from labelbox.schema.ontology import Ontology, OntologyBuilder, Classification, Option, Tool
 from labelbox.schema.role import Role, ProjectRole
 from labelbox.schema.invite import Invite, InviteLimit

labelbox/schema/__init__.py

@@ -12,7 +12,6 @@
 import labelbox.schema.task
 import labelbox.schema.user
 import labelbox.schema.webhook
-import labelbox.schema.prediction
 import labelbox.schema.ontology
 import labelbox.schema.invite
 import labelbox.schema.role

labelbox/schema/data_row.py

@@ -25,7 +25,6 @@ class DataRow(DbObject, Updateable, BulkDeletable):
         labels (Relationship): `ToMany` relationship to Label
         attachments (Relationship) `ToMany` relationship with AssetAttachment
         metadata (Relationship): This Relationship is Deprecated. Please use `DataRow.attachments()` instead
-        predictions (Relationship): `ToMany` relationship to Prediction
     """
     external_id = Field.String("external_id")
     row_data = Field.String("row_data")
@@ -47,7 +46,6 @@ class DataRow(DbObject, Updateable, BulkDeletable):
         "`DataRow.metadata()` is deprecated. Use `DataRow.attachments()` instead."
     )
     attachments = Relationship.ToMany("AssetAttachment", False, "attachments")
-    predictions = Relationship.ToMany("Prediction", False)
 
     supported_meta_types = supported_attachment_types = {
         attachment_type.value

labelbox/schema/prediction.py

@@ -51,8 +51,6 @@ class Project(DbObject, Updateable, Deletable):
         labeling_parameter_overrides (Relationship): `ToMany` relationship to LabelingParameterOverride
         webhooks (Relationship): `ToMany` relationship to Webhook
         benchmarks (Relationship): `ToMany` relationship to Benchmark
-        active_prediction_model (Relationship): `ToOne` relationship to PredictionModel
-        predictions (Relationship): `ToMany` relationship to Prediction
         ontology (Relationship): `ToOne` relationship to Ontology
     """
     name = Field.String("name")
@@ -75,9 +73,6 @@ class Project(DbObject, Updateable, Deletable):
         "LabelingParameterOverride", False, "labeling_parameter_overrides")
     webhooks = Relationship.ToMany("Webhook", False)
     benchmarks = Relationship.ToMany("Benchmark", False)
-    active_prediction_model = Relationship.ToOne("PredictionModel", False,
-                                                 "active_prediction_model")
-    predictions = Relationship.ToMany("Prediction", False)
     ontology = Relationship.ToOne("Ontology", True)
 
     def members(self):

labelbox/schema/project.py

@@ -1,4 +1,5 @@
 import uuid
+
 import pytest
 
 from labelbox.schema.labeling_frontend import LabelingFrontend
@@ -115,7 +116,6 @@ def configured_project(client, project, ontology, dataset):
 def prediction_id_mapping(configured_project):
     #Maps tool types to feature schema ids
     ontology = configured_project.ontology().normalized
-    inferences = []
     datarows = [d for d in list(configured_project.datasets())[0].data_rows()]
     result = {}
 

tests/integration/bulk_import/conftest.py

@@ -1,4 +1,5 @@
 from labelbox import ProjectRole
+import pytest
 
 
 def test_org_invite(client, organization, environ, queries):
@@ -72,6 +73,9 @@ def test_project_invite(client, organization, project_pack, queries):
     queries.cancel_invite(client, invite.uid)
 
 
+@pytest.mark.skip(
+    "Unable to programatically create user without accepting an email invite. Add back once there is a workaround."
+)
 def test_member_management(client, organization, project, project_based_user):
     roles = client.get_roles()
     assert not len(list(project_based_user.projects()))

tests/integration/test_user_management.py

@@ -42,7 +42,7 @@
     labelbox.Project, labelbox.Dataset, labelbox.DataRow, labelbox.Label,
     labelbox.AssetMetadata, labelbox.LabelingFrontend, labelbox.Task,
     labelbox.Webhook, labelbox.User, labelbox.Organization, labelbox.Review,
-    labelbox.Prediction, labelbox.PredictionModel, LabelerPerformance
+    LabelerPerformance
 ]
 
 ERROR_CLASSES = [LabelboxError] + LabelboxError.__subclasses__()