diff --git a/libs/labelbox/src/labelbox/schema/foundry/foundry_client.py b/libs/labelbox/src/labelbox/schema/foundry/foundry_client.py index 315fc831c..74073f9eb 100644 --- a/libs/labelbox/src/labelbox/schema/foundry/foundry_client.py +++ b/libs/labelbox/src/labelbox/schema/foundry/foundry_client.py @@ -18,8 +18,8 @@ def _create_app(self, app: App) -> App: $name: String!, $modelId: ID!, $ontologyId: ID!, $description: String, $inferenceParams: Json!, $classToSchemaId: Json! ){{ createModelFoundryApp(input: {{ - name: $name - modelId: $modelId + name: $name + modelId: $modelId ontologyId: $ontologyId description: $description inferenceParams: $inferenceParams @@ -89,6 +89,7 @@ def run_app( "classToSchemaId": app.class_to_schema_id, "inferenceParams": app.inference_params, "ontologyId": app.ontology_id, + "modelAppId": app.id, } data_rows_key = ( diff --git a/libs/labelbox/src/labelbox/schema/model_run.py b/libs/labelbox/src/labelbox/schema/model_run.py index 6ab816772..a7712858e 100644 --- a/libs/labelbox/src/labelbox/schema/model_run.py +++ b/libs/labelbox/src/labelbox/schema/model_run.py @@ -56,6 +56,7 @@ class ModelRun(DbObject): created_by_id = Field.String("created_by_id", "createdBy") model_id = Field.String("model_id") training_metadata = Field.Json("training_metadata") + model_app_id = Field.String("model_app_id") class Status(Enum): EXPORTING_DATA = "EXPORTING_DATA"