colin/colibre
1
0
Fork 0
Code Issues 30 Pull Requests Actions Packages Projects Releases 33 Activity

Correction de la recherche #67

Browse Source
This commit is contained in:
Colin Maudry
2026-01-29 21:48:13 +01:00
parent fe7277ce26
commit 776aa271f1
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/utils.py
+2 -1
View File
@@ -551,7 +551,6 @@ def search_org(dff: pl.DataFrame, query: str, org_type: str) -> pl.DataFrame:
dff = ( dff = (
dff.with_columns(token_matches + [match_score]) dff.with_columns(token_matches + [match_score])
.filter(pl.col("match_score") == len(tokens)) .filter(pl.col("match_score") == len(tokens))
.sort("Marchés", descending=True)
.drop([f"token_{token}" for token in tokens]) .drop([f"token_{token}" for token in tokens])
) )
@@ -567,6 +566,8 @@ def search_org(dff: pl.DataFrame, query: str, org_type: str) -> pl.DataFrame:
) )
dff = dff.select(f"{org_type}_id", f"{org_type}_nom", "Département", "Marchés") dff = dff.select(f"{org_type}_id", f"{org_type}_nom", "Département", "Marchés")
dff = dff.group_by(f"{org_type}_id", f"{org_type}_nom", "Département").sum()
dff = dff.sort("Marchés", descending=True)
return dff return dff