fix: default id stitchers not showing up in audit model

.github/workflows/tests.yaml

@@ -30,7 +30,8 @@ jobs:
       run: black --check . 
 
     - name: Install dependencies
-      run: pip install -r tests/integration/requirements.txt
+      working-directory: src/predictions
+      run: pip3 install .
 
     - name: Compile core train
       run: python -m src.predictions.profiles_mlcorelib.ml_core.preprocess_and_train --mode ci

src/predictions/profiles_mlcorelib/py_native/id_stitcher/audit.py

@@ -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

src/predictions/setup.py

@@ -32,6 +32,7 @@
     "networkx",
     "pyvis",
     "ruamel.yaml>=0.18.6",
+    "pytest",
 ]
 
 setup(