Skip to content

Commit

Permalink
fix: default id stitchers not showing up in audit model
Browse files Browse the repository at this point in the history
  • Loading branch information
shekhar-rudder committed Nov 12, 2024
1 parent f878ff5 commit 7d247d8
Showing 1 changed file with 7 additions and 3 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -57,11 +57,15 @@ def _set_id_stitcher_model(self, this: WhtMaterial):
if self.id_stitcher_model is not None:
return
id_stitcher_models = {}
models = this.base_wht_project.models(model_types=["id_stitcher"])
if len(models) == 0:
raise ValueError("No id_stitcher model found in the project")
# FIXME: Remove "identity" from the list once the ListModels bug is fixed in wht code
models = this.base_wht_project.models(model_types=["identity", "id_stitcher"])
for model in models:
# FIXME: Remove "id_collator" check once "identity" model type filter is removed
if model.model_type() == "id_collator":
continue
id_stitcher_models[model.name()] = model
if len(id_stitcher_models) == 0:
raise ValueError("No id_stitcher model found in the project")
if len(id_stitcher_models) > 1:
max_retries = 5
retry_count = 0
Expand Down

0 comments on commit 7d247d8

Please sign in to comment.