Skip to content

Commit c5025f1

Browse files
authored
add model app id to the model run creation and return it (#1973)
2 parents 2cebb8d + e23122e commit c5025f1

File tree

2 files changed

+4
-2
lines changed

2 files changed

+4
-2
lines changed

libs/labelbox/src/labelbox/schema/foundry/foundry_client.py

+3-2
Original file line numberDiff line numberDiff line change
@@ -18,8 +18,8 @@ def _create_app(self, app: App) -> App:
1818
$name: String!, $modelId: ID!, $ontologyId: ID!, $description: String, $inferenceParams: Json!, $classToSchemaId: Json!
1919
){{
2020
createModelFoundryApp(input: {{
21-
name: $name
22-
modelId: $modelId
21+
name: $name
22+
modelId: $modelId
2323
ontologyId: $ontologyId
2424
description: $description
2525
inferenceParams: $inferenceParams
@@ -89,6 +89,7 @@ def run_app(
8989
"classToSchemaId": app.class_to_schema_id,
9090
"inferenceParams": app.inference_params,
9191
"ontologyId": app.ontology_id,
92+
"modelAppId": app.id,
9293
}
9394

9495
data_rows_key = (

libs/labelbox/src/labelbox/schema/model_run.py

+1
Original file line numberDiff line numberDiff line change
@@ -56,6 +56,7 @@ class ModelRun(DbObject):
5656
created_by_id = Field.String("created_by_id", "createdBy")
5757
model_id = Field.String("model_id")
5858
training_metadata = Field.Json("training_metadata")
59+
model_app_id = Field.String("model_app_id")
5960

6061
class Status(Enum):
6162
EXPORTING_DATA = "EXPORTING_DATA"

0 commit comments

Comments
 (0)