also sort datatypes by depth

ontodev · Nov 16, 2023 · ea0630a · ea0630a
1 parent 81d973a
commit ea0630a
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/scripts/guess.py b/scripts/guess.py
@@ -242,9 +242,9 @@ def tiebreak(datatypes):
                     in_types.append(dt)
                 else:
                     other_types.append(dt)
-            sorted_types = sorted(in_types, key=lambda k: k["success_rate"], reverse=True) + sorted(
-                other_types, key=lambda k: k["success_rate"], reverse=True
-            )
+            sorted_types = sorted(
+                in_types, key=lambda k: (k["depth"], k["success_rate"]), reverse=True
+            ) + sorted(other_types, key=lambda k: (k["depth"], k["success_rate"]), reverse=True)
             return sorted_types[0]["datatype"]
 
         curr_index = 0
@@ -264,6 +264,7 @@ def tiebreak(datatypes):
                     matching_datatypes.append(
                         {
                             "datatype": datatype,
+                            "depth": curr_index,
                             "success_rate": success_rate,
                         }
                     )