Merge pull request #677 from Labelbox/mmw/AL-3352

Modify get_config return value [AL-3352]

Author: kkim-labelbox

CHANGELOG.md

@@ -1,6 +1,9 @@
 # Changelog
 
 # Version 0.0.0 (YYYY-MM-DD)
+### Changed
+* `ModelRun.get_config()`
+    * Modifies get_config to return un-nested Model Run config
 ### Added
 * `ModelRun.update_config()`
     * Updates model run training metadata

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(

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):