json dumps

Author: mmarinwilcke

labelbox/schema/model.py

@@ -1,4 +1,5 @@
 from typing import TYPE_CHECKING
+import json
 from labelbox.orm import query
 from labelbox.orm.db_object import DbObject
 from labelbox.orm.model import Entity, Field, Relationship
@@ -37,7 +38,7 @@ def create_model_run(self, name, config=None) -> "ModelRun":
             model_id_param, query.results_query_part(ModelRun))
         res = self.client.execute(query_str, {
             name_param: name,
-            config_param: config,
+            config_param: json.dumps(config),
             model_id_param: self.uid
         })
         return ModelRun(self.client, res["createModelRun"])

tests/integration/annotation_import/test_model_run.py

@@ -14,10 +14,11 @@ def test_model_run(client, configured_project_with_label, rand_gen):
     model = client.create_model(data["name"], data["ontology_id"])
 
     name = rand_gen(str)
-    config = {"batch_size": 100}
+    config = {"batch_size": 100, "reruns": None }
     model_run = model.create_model_run(name, config)
     assert model_run.name == name
     assert model_run.training_metadata["batchSize"] == config["batch_size"]
+    assert model_run.training_metadata["reruns"] == config["reruns"]
     assert model_run.model_id == model.uid
     assert model_run.created_by_id == client.get_user().uid