From 49abb70b59c28eceefbfa0c335d5a265d9c92f60 Mon Sep 17 00:00:00 2001 From: Al Date: Fri, 2 Oct 2015 20:24:21 -0400 Subject: [PATCH] [fix] dictionary --- scripts/geodata/osm/osm_address_training_data.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/scripts/geodata/osm/osm_address_training_data.py b/scripts/geodata/osm/osm_address_training_data.py index a4d136b7..1ccde02f 100644 --- a/scripts/geodata/osm/osm_address_training_data.py +++ b/scripts/geodata/osm/osm_address_training_data.py @@ -492,14 +492,16 @@ def build_address_format_training_data(language_rtree, infile, out_dir, tag_comp formatted_addresses = [] formatted_addresses.append(formatted_address) - address_components = {k: v for k, v in address_components.iteritems() if k in OSM_ADDRESS_COMPONENTS_VALID} + address_components = {k: v for k, v in address_components.iteritems() if k in OSM_ADDRESS_COMPONENT_VALUES} + if not address_components: + continue current_components = component_bitset(address_components.keys()) for component in address_components.keys(): - if current_components ^ OSM_ADDRESS_COMPONENTS_VALID[component] and random.random() >= 0.5: + if current_components ^ OSM_ADDRESS_COMPONENT_VALUES[component] and random.random() >= 0.5: address_components.pop(component) - current_components ^= OSM_ADDRESS_COMPONENTS_VALID[component] + current_components ^= OSM_ADDRESS_COMPONENT_VALUES[component] if not address_components: break