Add item for code_commune_insee label detection

datagouv · Feb 11, 2022 · 7c9c918 · 7c9c918
1 parent 2f9ffc7
commit 7c9c918
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/csv_detective/detect_labels/FR/geo/code_commune_insee/__init__.py b/csv_detective/detect_labels/FR/geo/code_commune_insee/__init__.py
@@ -6,10 +6,10 @@
 def _is(header):
     '''Returns 1 if the (processed) header matches one of the expected words combination, else 0'''
 
-    words_combinations_list = ['code commune insee', 'code insee', 'codes insee']
+    words_combinations_list = ['code commune insee', 'code insee', 'codes insee', 'code commune']
     processed_header = _process_text(header)
 
     header_matches_words_combination = float(any([words_combination == processed_header for words_combination in words_combinations_list]))
     words_combination_in_header = 0.5*float(any([full_word_strictly_inside_string(words_combination, processed_header) for words_combination in words_combinations_list]))
 
-    return max(header_matches_words_combination, words_combination_in_header)
+    return max(header_matches_words_combination, words_combination_in_header)