[fix] .get

2015-12-05 16:13:15 -05:00
parent c92a6de477
commit 3a7ba0288f
1 changed files with 3 additions and 1 deletions
--- a/scripts/geodata/osm/osm_address_training_data.py
+++ b/scripts/geodata/osm/osm_address_training_data.py
@@ -726,7 +726,9 @@ def build_address_format_training_data(admin_rtree, language_rtree, neighborhood
        Probabilistically strip standard prefixes/suffixes e.g. "London Borough of"
        '''
        for component in REPLACE_COMPONENTS:
-            name = address_components[component]
+            name = address_components.get(component)
+            if not name:
+                continue
            replacement = replace_name_prefixes(replace_name_suffixes(name))
            if replacement != name and random.random() < 0.6:
                address_components[component] = replacement