Default to 'float' if no point_type is given.
@@ -105,6 +105,8 @@ def main():
dataset = get_dataset(args.dataset)
dimension = len(dataset['train'][0]) # TODO(erikbern): ugly
+ point_type = 'float'
+ if 'point_type' in dataset.attrs:
point_type = dataset.attrs['point_type']
distance = dataset.attrs['distance']
definitions = get_definitions(args.definitions, dimension, point_type, distance, args.count)
