Merge pull request #483 from Labelbox/al-1677

[AL-1677] Client Get Model Run

Author: jtsodapop

labelbox/client.py

@@ -25,6 +25,7 @@
 from labelbox.schema import role
 from labelbox.schema.labeling_frontend import LabelingFrontend
 from labelbox.schema.model import Model
+from labelbox.schema.model_run import ModelRun
 from labelbox.schema.ontology import Ontology, Tool, Classification
 from labelbox.schema.organization import Organization
 from labelbox.schema.user import User
@@ -906,3 +907,15 @@ def create_feature_schema(self, normalized):
         # But the features are the same so we just grab the feature schema id
         res['id'] = res['normalized']['featureSchemaId']
         return Entity.FeatureSchema(self, res)
+
+    def get_model_run(self, model_run_id: str) -> ModelRun:
+        """ Gets a single ModelRun with the given ID.
+
+            >>> model_run = client.get_model_run("<model_run_id>")
+
+        Args:
+            model_run_id (str): Unique ID of the ModelRun.
+        Returns:
+            A ModelRun object.
+        """
+        return self._get_single(Entity.ModelRun, model_run_id)

tests/integration/annotation_import/test_model_run.py

@@ -23,6 +23,9 @@ def test_model_run(client, configured_project_with_label, rand_gen):
     assert model_run_data_row.data_row().uid == next(
         next(project.datasets()).data_rows()).uid
 
+    fetch_model_run = client.get_model_run(model_run.uid)
+    assert fetch_model_run == model_run
+
 
 def test_model_run_delete(client, model_run):
     models_before = list(client.get_models())