Skip to content

Commit 75c2d16

Browse files
committed
formatting
1 parent 7f01dc5 commit 75c2d16

File tree

2 files changed

+7
-1
lines changed

2 files changed

+7
-1
lines changed

labelbox/schema/model.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -41,6 +41,12 @@ def create_model_run(self, name, config=None) -> "ModelRun":
4141
config_param: json.dumps(config),
4242
model_id_param: self.uid
4343
})
44+
res = self.client.execute(
45+
query_str, {
46+
name_param: name,
47+
config_param: json.dumps(config),
48+
model_id_param: self.uid
49+
})
4450
return ModelRun(self.client, res["createModelRun"])
4551

4652
def delete(self) -> None:

tests/integration/annotation_import/test_model_run.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,7 @@ def test_model_run(client, configured_project_with_label, rand_gen):
1414
model = client.create_model(data["name"], data["ontology_id"])
1515

1616
name = rand_gen(str)
17-
config = {"batch_size": 100, "reruns": None }
17+
config = {"batch_size": 100, "reruns": None}
1818
model_run = model.create_model_run(name, config)
1919
assert model_run.name == name
2020
assert model_run.training_metadata["batchSize"] == config["batch_size"]

0 commit comments

Comments
 (0)