Addresses review comments

Author: mmarinwilcke

CHANGELOG.md

@@ -9,6 +9,19 @@
 * `RAW_TEXT` and `TEXT_FILE` attachment types to replace the `TEXT` type.
 
 # Version 3.25.3 (2022-08-10)
+
+### Added
+* `ModelRun.update_config()`
+    * Updates model run training metadata
+* `ModelRun.reset_config()`
+    * Resets model run training metadata
+* `ModelRun.get_config()`
+    * Fetches model run training metadata
+    
+### Changed
+* `Model.create_model_run()`
+    * Add training metadata config as a model run creation param
+
 ## Fixed
 * Label export will continue polling if the downloadUrl is None
 
@@ -663,7 +676,3 @@ a `Label`. Default value is 0.0.
 
 ## Version 2.2 (2019-10-18)
 Changelog not maintained before version 2.2.
-
-### Changed
-* `Model.create_model_run()`
-    * Add training metadata config as a model run creation param

labelbox/schema/model_run.py

@@ -263,7 +263,14 @@ def update_status(self,
             experimental=True)
 
     @experimental
-    def update_config(self, config: Dict[str, Any]):
+    def update_config(self, config: Dict[str, Any]) -> Dict[str, Any]:
+        """ 
+         Updates the model run's training metadata config
+         Args: 
+             config (dict): A dictionary of keys and values
+         Returns: 
+             model run id and updated training metadata
+         """ 
         data: Dict[str, Any] = {'config': config}
         res = self.client.execute(
             """mutation updateModelRunConfigPyApi($modelRunId: ID!, $data: UpdateModelRunConfigInput!){
@@ -277,7 +284,12 @@ def update_config(self, config: Dict[str, Any]):
         return res["updateModelRunConfig"]
 
     @experimental
-    def reset_config(self):
+    def reset_config(self) -> Dict[str, Any]:
+        """ 
+         Resets model run's training metadata config
+         Returns: 
+             Model run id and reset training metadata
+         """ 
         res = self.client.execute(
             """mutation resetModelRunConfigPyApi($modelRunId: ID!){
                 resetModelRunConfig(modelRun: {id : $modelRunId}){trainingMetadata}
@@ -287,7 +299,12 @@ def reset_config(self):
         return res["resetModelRunConfig"]
 
     @experimental
-    def fetch_config(self):
+    def get_config(self) -> Dict[str, Any]:
+        """ 
+         Gets model run's training metadata 
+         Returns: 
+             training metadata as a dictionary 
+         """ 
         res = self.client.execute("""query ModelRunPyApi($modelRunId: ID!){
                 modelRun(where: {id : $modelRunId}){trainingMetadata}
             }

tests/integration/annotation_import/test_model_run.py

@@ -67,10 +67,10 @@ def test_model_run_reset_config(model_run_with_training_metadata):
     assert res["trainingMetadata"] is None
 
 
-def test_model_run_fetch_config(model_run_with_training_metadata):
+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.fetch_config()
+    res = model_run_with_training_metadata.get_config()
     assert res["trainingMetadata"]["batchSize"] == new_config["batch_size"]