Modify return get_config return value [AL-3352]

mmarinwilcke · mmarinwilcke · commit 2d1065feb274 · 2022-08-23T16:19:49.000+08:00
diff --git a/labelbox/schema/model_run.py b/labelbox/schema/model_run.py
@@ -310,7 +310,7 @@ def get_config(self) -> Dict[str, Any]:
             }
         """, {'modelRunId': self.uid},
                                   experimental=True)
-        return res["modelRun"]
+        return res["modelRun"]["trainingMetadata"]
 
     @experimental
     def export_labels(
diff --git a/tests/integration/annotation_import/test_model_run.py b/tests/integration/annotation_import/test_model_run.py
@@ -71,7 +71,7 @@ def test_model_run_get_config(model_run_with_training_metadata):
     new_config = {"batch_size": 2000}
     model_run_with_training_metadata.update_config(new_config)
     res = model_run_with_training_metadata.get_config()
-    assert res["trainingMetadata"]["batch_size"] == new_config["batch_size"]
+    assert res["batch_size"] == new_config["batch_size"]
 
 
 def test_model_run_data_rows_delete(client, model_run_with_model_run_data_rows):

Original file line number	Diff line number	Diff line change
`@@ -310,7 +310,7 @@ def get_config(self) -> Dict[str, Any]:`
`310`	`310`	`}`
`311`	`311`	`""", {'modelRunId': self.uid},`
`312`	`312`	`experimental=True)`
`313`		`- return res["modelRun"]`
	`313`	`+ return res["modelRun"]["trainingMetadata"]`
`314`	`314`
`315`	`315`	`@experimental`
`316`	`316`	`def export_labels(`